Commit 42f10617 authored by Yuxin Wu's avatar Yuxin Wu

Set both starting_epoch and epoch_num from json (fix #473)

parent d44a1a20
......@@ -261,6 +261,7 @@ class JSONWriter(TrainingMonitor):
# TODO is this a good idea?
logger.info("Found training history from JSON, now starting from epoch number {}.".format(epoch))
self.trainer.loop.starting_epoch = epoch
self.trainer.loop._epoch_num = epoch - 1
else:
self._stats = []
self._stat_now = {}
......
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