| from .ret_msrvtt import * | |
| train_file = [ | |
| f"{anno_root_downstream}/ssv2_ret_template_train.json", | |
| f"{data_root}/ssv2", | |
| "video", | |
| ] | |
| test_file = dict( | |
| val=[ | |
| f"{anno_root_downstream}/ssv2_ret_template_val_small.json", | |
| f"{data_root}/ssv2", | |
| "video", | |
| ], | |
| ) | |
| test_types = ["val"] | |
| stop_key = None # used to choose the best ckpt. If None, save the last. | |
| has_multi_vision_gt = True | |
| scheduler["epochs"] = 10 | |
| optimizer["lr"] = 1e-4 | |
| max_txt_l = 22 | |