From 30b8dd063e2889a5559c67827ecc7b68920c6495 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Tue, 27 Sep 2022 15:54:37 +0100 Subject: [PATCH] fixup --- aimodel/src/lib/ai/RainfallWaterContraster.py | 1 + 1 file changed, 1 insertion(+) diff --git a/aimodel/src/lib/ai/RainfallWaterContraster.py b/aimodel/src/lib/ai/RainfallWaterContraster.py index 3196dc7..7451dfe 100644 --- a/aimodel/src/lib/ai/RainfallWaterContraster.py +++ b/aimodel/src/lib/ai/RainfallWaterContraster.py @@ -14,6 +14,7 @@ from .model_rainfallwater_contrastive import model_rainfallwater_contrastive from .helpers import make_callbacks from .helpers import summarywriter from .components.LayerContrastiveEncoder import LayerContrastiveEncoder +from .components.LayerConvNeXtGamma import LayerConvNeXtGamma from .components.LayerCheeseMultipleOut import LayerCheeseMultipleOut from .helpers.summarywriter import summarywriter