From 1bc8a5bf1374710a73937c520dcffeac2d230e6e Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 15 Sep 2022 19:37:06 +0100 Subject: [PATCH] ai: fix crash --- aimodel/src/subcommands/pretrain_predict.py | 1 - 1 file changed, 1 deletion(-) diff --git a/aimodel/src/subcommands/pretrain_predict.py b/aimodel/src/subcommands/pretrain_predict.py index 3298728..84fed70 100644 --- a/aimodel/src/subcommands/pretrain_predict.py +++ b/aimodel/src/subcommands/pretrain_predict.py @@ -53,7 +53,6 @@ def run(args): dataset = dataset_predict( dirpath_input=args.input, - batch_size=ai.batch_size, parallel_reads_multiplier=args.read_multiplier )