File size: 651 Bytes
60c8e7f |
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 26 27 28 |
from .qa import *
train_file = [
[
f"{anno_root_downstream}/msrvtt_qa_train.json",
f"{data_root}/msrvtt_2fps_224",
"video",
]
]
test_file = dict(
val=[
f"{anno_root_downstream}/msrvtt_qa_val.json",
f"{data_root}/msrvtt_2fps_224",
"video",
],
test=[
f"{anno_root_downstream}/msrvtt_qa_test.json",
f"{data_root}/msrvtt_2fps_224",
"video",
],
)
dataset_name = "msrvtt"
answer_list = f"{anno_root_downstream}/msrvtt_qa_answer_list.json" # list of answer words
test_types = ["val"]
stop_key = "val" # used to choose the best ckpt. If None, save the last.
|