1
0
Fork 0
mirror of https://github.com/prise6/smart-iss-posts synced 2024-05-01 21:32:43 +02:00

log version keras

This commit is contained in:
prise6 2019-03-10 17:07:41 +01:00
parent 4e4fc39b69
commit 7c8922940a

View file

@ -3,7 +3,7 @@
from keras.callbacks import ModelCheckpoint, CSVLogger from keras.callbacks import ModelCheckpoint, CSVLogger
from iss.models.Callbacks import DisplayPictureCallback from iss.models.Callbacks import DisplayPictureCallback
from iss.tools.tools import Tools from iss.tools.tools import Tools
import keras
class ModelTrainer: class ModelTrainer:
@ -25,6 +25,8 @@ class ModelTrainer:
def train(self): def train(self):
print(keras.__version__)
self.model.model.fit_generator( self.model.model.fit_generator(
generator = self.data_loader.get_train_generator(), generator = self.data_loader.get_train_generator(),
steps_per_epoch = self.steps_per_epoch, steps_per_epoch = self.steps_per_epoch,