From 026c9f519d1da91d5f8d2c0656c2c9f1f247d944 Mon Sep 17 00:00:00 2001
From: shippingwang <shipeng1108@163.com>
Date: Mon, 29 Jun 2020 12:23:30 +0000
Subject: [PATCH] fix

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

diff --git a/tools/train.py b/tools/train.py
index c7ae7320..607f4c69 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -133,7 +133,7 @@ def main(args):
 
                 top1_acc = program.run(valid_dataloader, exe,
                                        compiled_valid_prog, valid_fetchs,
-                                       epoch_id, 'valid')
+                                       epoch_id, 'valid', config)
                 if top1_acc > best_top1_acc:
                     best_top1_acc = top1_acc
                     message = "The best top1 acc {:.5f}, in epoch: {:d}".format(
-- 
GitLab