Make eta respect `log_smooth_window` setting (#3580)

ac103510 · Yang Zhang · GitHub · 97fc087b · ac103510
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

tools/train.py tools/train.py +1 -1

未找到文件。
--- a/tools/train.py
+++ b/tools/train.py
@@ -221,7 +221,7 @@ def main():

    cfg_name = os.path.basename(FLAGS.config).split('.')[0]
    save_dir = os.path.join(cfg.save_dir, cfg_name)
-    time_stat = deque(maxlen=cfg.log_iter)
+    time_stat = deque(maxlen=cfg.log_smooth_window)
    best_box_ap_list = [0.0, 0]  #[map, iter]

    # use tb-paddle to log data