Merge pull request #272 from MannyKayy/patch-1

ChainerV2 calls trigger outside of snapshot object
chainer · Jun 13, 2017 · aad4f55 · aad4f55
2 parents 78be7c1 + 1ccaa7e
commit aad4f55
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/segnet/train.py b/examples/segnet/train.py
@@ -94,8 +94,8 @@ def transform(in_data):
             file_name='loss.png'))
 
     trainer.extend(extensions.snapshot_object(
-        model.predictor, filename='model_iteration-{.updater.iteration}',
-        trigger=end_trigger))
+        model.predictor, filename='model_iteration-{.updater.iteration}'),
+        trigger=end_trigger)
     trainer.extend(extensions.PrintReport(
         ['epoch', 'iteration', 'elapsed_time', 'lr',
          'main/loss', 'validation/main/loss']),