From 1ccaa7eddd9d5c9116859bac4bbf4f04d692651b Mon Sep 17 00:00:00 2001 From: Emmanuel Kahembwe Date: Tue, 13 Jun 2017 17:54:12 +0100 Subject: [PATCH] ChainerV2 calls trigger outside of snapshot object --- examples/segnet/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/segnet/train.py b/examples/segnet/train.py index 80f7229acc..8b21278d7c 100644 --- 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']),