|
|
@ -163,7 +163,7 @@ class Trainer():
|
|
|
|
"epoch": self.epoch,
|
|
|
|
"epoch": self.epoch,
|
|
|
|
"lr": self.optimizer.get_lr()
|
|
|
|
"lr": self.optimizer.get_lr()
|
|
|
|
})
|
|
|
|
})
|
|
|
|
self.checkpoint.add_checkpoint(self.checkpoint_dir, self.iteration
|
|
|
|
self.checkpoint.save_parameters(self.checkpoint_dir, self.iteration
|
|
|
|
if tag is None else tag, self.model,
|
|
|
|
if tag is None else tag, self.model,
|
|
|
|
self.optimizer, infos)
|
|
|
|
self.optimizer, infos)
|
|
|
|
|
|
|
|
|
|
|
|