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
 	)