From c52a9f961c5b2f2e06645e8bf0f4907733b75bad Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 31 Aug 2022 17:37:28 +0100 Subject: [PATCH] and another --- aimodel/src/lib/ai/RainfallWaterContraster.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/RainfallWaterContraster.py b/aimodel/src/lib/ai/RainfallWaterContraster.py index b452799..c871949 100644 --- a/aimodel/src/lib/ai/RainfallWaterContraster.py +++ b/aimodel/src/lib/ai/RainfallWaterContraster.py @@ -57,7 +57,7 @@ class RainfallWaterContraster(object): def make_model(self): - model = model_rainfallwater_contrastive(batch_size=self.batch_size, **self.kwargs) + model = model_rainfallwater_contrastive(**self.kwargs) return model