Fix post proess for AMR

line · Sep 20, 2024 · f043607 · f043607
1 parent 3b23561
commit f043607
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/training/evaluate.py b/training/evaluate.py
@@ -252,13 +252,13 @@ def compute_mr_results(epoch_i, model, eval_loader, opt, criterion=None):
             min_w_l=2, max_w_l=150, move_window_method="left",
             process_func_names=("clip_ts", "round_multiple")
         )
-    elif opt.dset_name == 'charades':
+    elif opt.dset_name in ['charades', 'clotho-moment', 'unav100', 'tut2017']:
         post_processor = PostProcessorDETR(
             clip_length=opt.clip_length, min_ts_val=0, max_ts_val=150,
             min_w_l=2, max_w_l=60, move_window_method="left",
             process_func_names=("clip_ts", "round_multiple")
         )
-    elif opt.dset_name in ['tacos', 'activitynet', 'youtube_highlight', 'clotho-moment', 'unav100', 'tut2017']:
+    elif opt.dset_name in ['tacos', 'activitynet', 'youtube_highlight']:
         post_processor = PostProcessorDETR(
             clip_length=opt.clip_length, min_ts_val=0, max_ts_val=50000,
             min_w_l=0, max_w_l=50000, move_window_method="left",