From 69a2d0cf04ec9424c06d4d85e61bc388710f7e3a Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 11 Nov 2022 18:27:01 +0000 Subject: [PATCH] fixup --- aimodel/src/subcommands/train_mono.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/aimodel/src/subcommands/train_mono.py b/aimodel/src/subcommands/train_mono.py index bc8a9da..bccbe03 100644 --- a/aimodel/src/subcommands/train_mono.py +++ b/aimodel/src/subcommands/train_mono.py @@ -54,8 +54,7 @@ def run(args): dirpath_input=args.input, batch_size=args.batch_size, water_threshold=args.water_threshold, - shape_water_desired=[args.water_size, args.water_size], - dummy_label=False + shape_water_desired=[args.water_size, args.water_size] ) dataset_metadata = read_metadata(args.input)