fix bug of resumed iteration
@@ -9,8 +9,8 @@ from chainer.training import triggers
class DummyUpdater(training.Updater):
- def __init__(self, iters_per_epoch, initial_iteration=0):
- self.iteration = initial_iteration
+ def __init__(self, iters_per_epoch, initial_iteration=1):
+ self.iteration = initial_iteration - 1
self.iters_per_epoch = iters_per_epoch
def finalize(self):
