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",