fix for non serialisable models

jkiesele · jkiesele · commit bcc95cce7571 · 2022-01-18T21:20:04.000+01:00
diff --git a/training/training_base.py b/training/training_base.py
@@ -485,7 +485,7 @@ def trainModel(self,
                 traingen.shuffleFileList()
                 #
         
-            self.saveModel("KERAS_model")
+            self.saveModel("KERAS_model.h5")
 
         return self.keras_model, self.callbacks.history
     

Original file line number	Diff line number	Diff line change
`@@ -485,7 +485,7 @@ def trainModel(self,`
`485`	`485`	`traingen.shuffleFileList()`
`486`	`486`	`#`
`487`	`487`
`488`		`- self.saveModel("KERAS_model")`
	`488`	`+ self.saveModel("KERAS_model.h5")`
`489`	`489`
`490`	`490`	`return self.keras_model, self.callbacks.history`
`491`	`491`