From 9b25186541272704774dbe4b77cc0bce8f8815a4 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 14 Sep 2022 15:55:21 +0100 Subject: [PATCH] fix --only-gpu --- aimodel/src/subcommands/pretrain_plot.py | 2 -- aimodel/src/subcommands/pretrain_predict.py | 2 -- 2 files changed, 4 deletions(-) diff --git a/aimodel/src/subcommands/pretrain_plot.py b/aimodel/src/subcommands/pretrain_plot.py index 33a355c..fd41f92 100644 --- a/aimodel/src/subcommands/pretrain_plot.py +++ b/aimodel/src/subcommands/pretrain_plot.py @@ -15,8 +15,6 @@ def parse_args(): # parser.add_argument("--config", "-c", help="Filepath to the TOML config file to load.", required=True) parser.add_argument("--input", "-i", help="Path to input file containing the content to plot.", required=True) parser.add_argument("--output", "-o", help="Path to output file to write the resulting image to.", required=True) - parser.add_argument("--only-gpu", - help="If the GPU is not available, exit with an error (useful on shared HPC systems to avoid running out of memory & affecting other users)", action="store_true") return parser diff --git a/aimodel/src/subcommands/pretrain_predict.py b/aimodel/src/subcommands/pretrain_predict.py index 03140a1..258ee76 100644 --- a/aimodel/src/subcommands/pretrain_predict.py +++ b/aimodel/src/subcommands/pretrain_predict.py @@ -25,8 +25,6 @@ def parse_args(): parser.add_argument("--reads-multiplier", help="Optional. The multiplier for the number of files we should read from at once. Defaults to 1.5, which means read ceil(NUMBER_OF_CORES * 1.5). Set to a higher number of systems with high read latency to avoid starving the GPU of data.") parser.add_argument("--no-vis", help="Don't also plot a visualisation of the resulting embeddings.", action="store_true") - parser.add_argument("--only-gpu", - help="If the GPU is not available, exit with an error (useful on shared HPC systems to avoid running out of memory & affecting other users)", action="store_true") return parser