From 4cd8fc6dedfcd67623ea96739f45775193737646 Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Wed, 28 Sep 2022 17:37:42 +0100
Subject: [PATCH] segmentation: param name fix

---
 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 fc1da4e..3d0b214 100644
--- a/aimodel/src/subcommands/train.py
+++ b/aimodel/src/subcommands/train.py
@@ -52,7 +52,7 @@ def run(args):
 	ai = RainfallWaterSegmenter(
 		dir_output=args.output,
 		batch_size=args.batch_size,
-		feature_dim=args.feature_dim,
+		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.