Improve formatting line break
parent
02fa053653
commit
80b13db2e7
|
@ -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()
|
||||
|
|
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue