From 404dc30f08f56c37acac6eb6da26af556f0d8ea6 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 28 Sep 2022 17:39:09 +0100 Subject: [PATCH] and again --- aimodel/src/subcommands/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/train.py b/aimodel/src/subcommands/train.py index 3d0b214..0969451 100644 --- a/aimodel/src/subcommands/train.py +++ b/aimodel/src/subcommands/train.py @@ -55,7 +55,7 @@ def run(args): feature_dim_in=args.feature_dim, metadata = read_metadata(args.input), - shape_water=[ args.water_size, args.water_size ] # The DESIRED output shape. the actual data will be cropped to match this. + shape_water_out=[ args.water_size, args.water_size ] # The DESIRED output shape. the actual data will be cropped to match this. ) ai.train(dataset_train, dataset_validate)