From f2ae74ce7b3fa023644f670c9e32cfa26a918ffc Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 2 Nov 2022 17:38:26 +0000 Subject: [PATCH] how could I be so stupid..... round 2 --- aimodel/src/lib/ai/RainfallWaterSegmenter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/RainfallWaterSegmenter.py b/aimodel/src/lib/ai/RainfallWaterSegmenter.py index e592d37..dee79cd 100644 --- a/aimodel/src/lib/ai/RainfallWaterSegmenter.py +++ b/aimodel/src/lib/ai/RainfallWaterSegmenter.py @@ -82,7 +82,7 @@ class RainfallWaterSegmenter(object): validation_data=dataset_validate, epochs=self.epochs, callbacks=make_callbacks(self.dir_output, self.model), - steps_per_epoch=10 # For testing + # steps_per_epoch=10 # For testing ) def embed(self, rainfall_embed):