From deffe69202303f3b373f9787bcb237af27b422eb Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Mon, 3 Oct 2022 16:59:36 +0100 Subject: [PATCH] typo --- 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 5419d0b..4681e17 100644 --- a/aimodel/src/lib/ai/RainfallWaterSegmenter.py +++ b/aimodel/src/lib/ai/RainfallWaterSegmenter.py @@ -80,7 +80,7 @@ class RainfallWaterSegmenter(object): dataset_train, validation_data=dataset_validate, epochs=self.epochs, - callbacks=make_callbacks(self.dir_output, self.model_predict), + callbacks=make_callbacks(self.dir_output, self.model), steps_per_epoch=10 # For testing )