sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-12-22 22:25:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

fix --only-gpu

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-14 15:55:21 +01:00
parent a9c9c70d13
commit 9b25186541
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
2 changed files with 0 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/src/subcommands/pretrain_plot.py
View file

@ -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

2
aimodel/src/subcommands/pretrain_predict.py
View file

@ -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