From aa7d9b8cf6e9e9b51ecf63c238be7a04e4c82ba4 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 10 Nov 2022 19:46:09 +0000 Subject: [PATCH] fixup --- aimodel/src/subcommands/train_predict.py | 1 + 1 file changed, 1 insertion(+) diff --git a/aimodel/src/subcommands/train_predict.py b/aimodel/src/subcommands/train_predict.py index 4fed4b3..4eb9727 100644 --- a/aimodel/src/subcommands/train_predict.py +++ b/aimodel/src/subcommands/train_predict.py @@ -42,6 +42,7 @@ def run(args): 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") + return 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: