mirror of
https://github.com/prise6/smart-iss-posts
synced 2024-05-02 21:53:10 +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 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,
|
||||
|
|
Loading…
Reference in a new issue