From e7410fb480e6a181c48e126d7ca9b67babb77ef2 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 25 Nov 2022 17:47:17 +0000 Subject: [PATCH] train_mono_predict: limit label size to 64x64 that's the size the model predicts --- aimodel/src/subcommands/train_mono_predict.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/train_mono_predict.py b/aimodel/src/subcommands/train_mono_predict.py index 73bcd1a..1fdd333 100644 --- a/aimodel/src/subcommands/train_mono_predict.py +++ b/aimodel/src/subcommands/train_mono_predict.py @@ -74,7 +74,8 @@ def run(args): # 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_mono_predict( dirpath_input=args.input, - parallel_reads_multiplier=args.read_multiplier + parallel_reads_multiplier=args.read_multiplier, + shape_water_desired=[64, 64] ) # for items in dataset_train.repeat(10):