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

train_predict: add error message for parrams.json not found

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-11-10 19:45:41 +00:00
parent e519b0adb3
commit 0894bd09e8
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -40,6 +40,8 @@ def run(args):
if (not hasattr(args, "params")) or args.params == None:
args.params = find_paramsjson(args.checkpoint)
if args.params == None:
logger.error("Error: Failed to find params.json. Please ensure it's either in the same directory as the checkpoint or 1 level above")
if (not hasattr(args, "read_multiplier")) or args.read_multiplier == None:
args.read_multiplier = 0
if (not hasattr(args, "records_per_file")) or args.records_per_file == None: