From 1e1d6dd2738712cfc47ec3014793fb4de6f0144a Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 24 Nov 2022 17:48:19 +0000 Subject: [PATCH] fixup --- aimodel/src/subcommands/train_mono_predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/train_mono_predict.py b/aimodel/src/subcommands/train_mono_predict.py index ed0ab74..e413f47 100644 --- a/aimodel/src/subcommands/train_mono_predict.py +++ b/aimodel/src/subcommands/train_mono_predict.py @@ -72,7 +72,7 @@ def run(args): sys.stderr.write(f"\n\n>>> This is TensorFlow {tf.__version__}\n\n\n") # Note that if using a directory of input files, the output order is NOT GUARANTEED TO BE THE SAME. In fact, it probably won't be (see dataset_mono for more details). - dataset = dataset_predict( + dataset = dataset_mono_predict( dirpath_input=args.input, parallel_reads_multiplier=args.read_multiplier )