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:
parent
4e4fc39b69
commit
7c8922940a
|
@ -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,
|
||||||
|
|
Loading…
Reference in a new issue