This commit is contained in:
Starbeamrainbowlabs 2022-11-10 19:46:09 +00:00
parent 0894bd09e8
commit aa7d9b8cf6
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

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