From 4514086dc612be2956341e490b4aa4a2e03698dd Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 20 Jan 2023 18:39:35 +0000 Subject: [PATCH] make_encoderonly: kwargs --- aimodel/src/encoderonly_test_rainfall.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/encoderonly_test_rainfall.py b/aimodel/src/encoderonly_test_rainfall.py index c6d4355..8e7a657 100755 --- a/aimodel/src/encoderonly_test_rainfall.py +++ b/aimodel/src/encoderonly_test_rainfall.py @@ -60,7 +60,7 @@ dataset_train, dataset_validate = dataset_encoderonly( # ██ ██ ██ ██ ██ ██ ██ ██ ██ # ██ ██ ██████ ██████ ███████ ███████ -def make_encoderonly(windowsize, channels, encoder="convnext", water_bins=2): +def make_encoderonly(windowsize, channels, encoder="convnext", water_bins=2, **kwargs): if encoder == "convnext": model = make_convnext(input_shape=(windowsize, windowsize, channels), num_classes=water_bins, **kwargs) elif encoder == "resnet":