switch to a smaller ConvNeXt

2024-06-30 18:44:55 +00:00 · 2023-01-20 19:14:38 +00:00 · 2023-01-20 19:14:38 +00:00 · e72d3991b8
parent e1ad16a213
commit e72d3991b8
1 changed files with 1 additions and 0 deletions
--- a/aimodel/src/encoderonly_test_rainfall.py
+++ b/aimodel/src/encoderonly_test_rainfall.py
@ -67,6 +67,7 @@ def make_encoderonly(windowsize, channels, encoder="convnext", water_bins=2, **k
 			input_shape=(windowsize, windowsize, channels),
 			num_classes=water_bins,
 			downsample_at_start=False,
+			arch_name="convnext_xtiny",
 			**kwargs
 		)
 	elif encoder == "resnet":