From f0436079161a7bdf21305704ea7860f5b806c034 Mon Sep 17 00:00:00 2001 From: hokuto-munakata Date: Fri, 20 Sep 2024 02:15:21 +0900 Subject: [PATCH] Fix post proess for AMR --- training/evaluate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/training/evaluate.py b/training/evaluate.py index dcdd7de..9c89ede 100755 --- 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",