File size: 481 Bytes
401fa20 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
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
|