Improve formatting line break

pull/23/merge
Matthew Scholefield 2018-08-10 00:03:16 -05:00
parent 02fa053653
commit 80b13db2e7
2 changed files with 4 additions and 3 deletions

View File

@ -164,8 +164,8 @@ class Trainer:
train_inputs, train_outputs = self.sampled_data
self.model.fit(
train_inputs, train_outputs, self.args.batch_size,
self.epoch + self.args.epochs, validation_data=self.test, initial_epoch=self.epoch,
callbacks=self.callbacks
self.epoch + self.args.epochs, validation_data=self.test,
initial_epoch=self.epoch, callbacks=self.callbacks
)
except KeyboardInterrupt:
print()

View File

@ -98,10 +98,11 @@ class IncrementalTrainer(Trainer):
train_data = TrainData.merge(train_data, self.sampled_data)
test_data = TrainData.merge(test_data, self.test)
train_inputs, train_outputs = train_data
print()
try:
self.listener.runner.model.fit(
train_data[0], train_data[1], self.args.batch_size, self.epoch + self.args.epochs,
train_inputs, train_outputs, self.args.batch_size, self.epoch + self.args.epochs,
validation_data=test_data, callbacks=self.callbacks, initial_epoch=self.epoch
)
finally: