From c27869630a9db0988279734fcd94bdb28966b84a Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 8 Dec 2022 18:58:54 +0000 Subject: [PATCH] I hate VSCode's git commit interface it doesn't let you ammend --- aimodel/src/subcommands/train_mono.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/train_mono.py b/aimodel/src/subcommands/train_mono.py index 8f86745..5055812 100644 --- a/aimodel/src/subcommands/train_mono.py +++ b/aimodel/src/subcommands/train_mono.py @@ -79,7 +79,7 @@ def run(args): model_arch_enc = args.arch_enc, model_arch_dec = args.arch_dec, learning_rate = args.learning_rate, - heightmap_input = args.heightmap is not None + heightmap_input = args.heightmap is not None, metadata = read_metadata(args.input), # shape_water_out=[ args.water_size, args.water_size ], # The DESIRED output shape. the actual data will be cropped to match this.