From dbb981d322890ae8751351d7b051192ecb9bc34e Mon Sep 17 00:00:00 2001
From: yukavio <67678385+yukavio@users.noreply.github.com>
Date: Wed, 13 Jan 2021 10:42:10 +0800
Subject: [PATCH] fix demo/prune log (#598)

---
 demo/prune/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/demo/prune/train.py b/demo/prune/train.py
index 574f5158..da2c7eb5 100644
--- a/demo/prune/train.py
+++ b/demo/prune/train.py
@@ -239,7 +239,7 @@ def compress(args):
 
     for i in range(args.num_epochs):
         train(i, train_program)
-        if i % args.test_period == 0:
+        if (i + 1) % args.test_period == 0:
             test(i, pruned_val_program)
             save_model(exe, pruned_val_program,
                        os.path.join(args.model_path, str(i)))
-- 
GitLab