From 4202821d98bbb5d2ea14d368da4bbdcee8f2dfaa Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Tue, 6 Sep 2022 15:37:36 +0100 Subject: [PATCH] typo --- aimodel/src/subcommands/pretrain.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/pretrain.py b/aimodel/src/subcommands/pretrain.py index 6d5d199..865e8c9 100644 --- a/aimodel/src/subcommands/pretrain.py +++ b/aimodel/src/subcommands/pretrain.py @@ -55,7 +55,7 @@ def run(args): feature_dim=args.feature_dim, metadata = read_metadata(args.input), - shape_water=[ args.shape_water, args.shape_water ] # The DESIRED output shape. the actual data will be cropped to match this. + shape_water=[ 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)