rei-forks/precise-lite-amd64aarch64
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix Python 3.4 compatibility with precise-train-sampled

Browse Source
This commit is contained in:
Matthew D. Scholefield 2019-04-15 16:21:58 +00:00
parent 53720c6a99
commit e1211d0af0
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
precise/scripts/train_sampled.py
View File

@ -83,8 +83,10 @@ class SampledTrainer(Trainer):
self.write_sampling_metrics(predicted)
self.model.fit(
*self.sampled_data, self.args.batch_size, self.epoch + self.args.epochs,
callbacks=self.callbacks, initial_epoch=self.epoch, validation_data=self.test
*self.sampled_data, batch_size=self.args.batch_size,
epochs=self.epoch + self.args.epochs,
callbacks=self.callbacks, initial_epoch=self.epoch,
validation_data=self.test
)