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