From 3697a31eb8de891ddd7814cb7003319dd14f909f Mon Sep 17 00:00:00 2001
From: Liufang Sang <slf12thuss@163.com>
Date: Fri, 15 May 2020 00:17:46 -0500
Subject: [PATCH] Fix train_eval in quantization (#682)

---
 slim/quantization/train.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/slim/quantization/train.py b/slim/quantization/train.py
index 2e89dc09a..734dc39d4 100644
--- a/slim/quantization/train.py
+++ b/slim/quantization/train.py
@@ -256,8 +256,14 @@ def main():
 
             if FLAGS.eval:
                 # evaluation
-                results = eval_run(exe, compiled_eval_prog, eval_loader,
-                                   eval_keys, eval_values, eval_cls)
+                results = eval_run(
+                    exe,
+                    compiled_eval_prog,
+                    eval_loader,
+                    eval_keys,
+                    eval_values,
+                    eval_cls,
+                    cfg=cfg)
                 resolution = None
                 if 'mask' in results[0]:
                     resolution = model.mask_head.resolution
-- 
GitLab