From e72d3991b8155a72121523c3b7b50677e05bf3f8 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 20 Jan 2023 19:14:38 +0000 Subject: [PATCH] switch to a smaller ConvNeXt --- aimodel/src/encoderonly_test_rainfall.py | 1 + 1 file changed, 1 insertion(+) diff --git a/aimodel/src/encoderonly_test_rainfall.py b/aimodel/src/encoderonly_test_rainfall.py index 0b180bf..2925222 100755 --- 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":