From f4a2c742d9fc1b86f9c535daf616fc3b51c136ff Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Tue, 11 Oct 2022 19:19:23 +0100
Subject: [PATCH] typo

---
 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 999c594..dde76cc 100644
--- a/aimodel/src/subcommands/train.py
+++ b/aimodel/src/subcommands/train.py
@@ -65,7 +65,7 @@ def run(args):
 		batch_size=args.batch_size,
 		feature_dim_in=args.feature_dim,
 		
-		model_arch=args.model_arch,
+		model_arch=args.arch,
 		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.
 	)