diff --git a/slim/quantization/train.py b/slim/quantization/train.py
index 2e89dc09abd78bf87df0fd1b6829ca1060984f63..734dc39d47244b9cde1a63a55e38e9bd7a4c21ea 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