log version keras

This commit is contained in:
prise6 2019-03-10 17:07:41 +01:00
parent 4e4fc39b69
commit 7c8922940a
1 changed files with 3 additions and 1 deletions

View File

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