"docs/en/vscode:/vscode.git/clone" did not exist on "57ba4811f147a6ae721db58bd988a8dee6c284b3"
Commit 6a7c8d0d authored by alexeib's avatar alexeib Committed by Myle Ott
Browse files

address comments

parent fe54ea54
......@@ -57,13 +57,13 @@ class Trainer(object):
self._max_bsz_seen = 0
self._num_updates = 0
self._optim_history = None
def save_checkpoint(self, filename, extra_state):
"""Save all training state in a checkpoint file."""
if distributed_utils.is_master(self.args): # only save one checkpoint
optim_history = self._optim_history if hasattr(self, '_optim_history') else None
utils.save_state(filename, self.args, self.model, self.criterion, self.optimizer,
self.lr_scheduler, self._num_updates, optim_history, extra_state)
self.lr_scheduler, self._num_updates, self._optim_history, extra_state)
def load_checkpoint(self, filename):
"""Load all training state from a checkpoint file."""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment