We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 33c351e commit 5754a2cCopy full SHA for 5754a2c
onmt/trainer.py
@@ -308,13 +308,15 @@ def train(self,
308
if (self.model_saver is not None
309
and (save_checkpoint_steps != 0
310
and step % save_checkpoint_steps == 0)):
311
- self.model_saver.save(step, is_best, moving_average=self.moving_average)
+ self.model_saver.save(step, is_best,
312
+ moving_average=self.moving_average)
313
314
if train_steps > 0 and step >= train_steps:
315
break
316
317
if self.model_saver is not None:
318
319
320
return total_stats
321
322
def validate(self, valid_iter, moving_average=None):
0 commit comments