diff --git a/tools/train.py b/tools/train.py index 1a6250901d0db1a09cc098aa0fa79edcb5bc3726..27551c599d883b9c17059df2d3c72b914598dfcd 100644 --- 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