diff options
Diffstat (limited to 'lstm_chem/trainer.py')
| -rw-r--r--[-rwxr-xr-x] | lstm_chem/trainer.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/lstm_chem/trainer.py b/lstm_chem/trainer.py index 4e8057e..0ed608a 100755..100644 --- a/lstm_chem/trainer.py +++ b/lstm_chem/trainer.py @@ -32,7 +32,8 @@ class LSTMChemTrainer(object): )) def train(self): - history = self.model.fit_generator( +# history = self.model.fit_generator( + history = self.model.fit( self.train_data_loader, steps_per_epoch=self.train_data_loader.__len__(), epochs=self.config.num_epochs, |