This commit is contained in:
Starbeamrainbowlabs 2022-10-03 16:59:36 +01:00
parent fc6d2dabc9
commit deffe69202
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -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
)