vtt.yaml 658 Bytes
Newer Older
huaerkl's avatar
v1.0  
huaerkl committed
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
includes: projects/task/ft.yaml
dataset:
  meta_processor: MSRVTTMetaProcessor
  train_path: data/msrvtt/MSRVTT_train.csv
  jsfusion_path: data/msrvtt/MSRVTT_JSFUSION_test.csv
  full_test_path: data/msrvtt/MSRVTT_FULL_test.csv
  dup: 20
  val_path: data/msrvtt/MSRVTT_JSFUSION_test.csv
  vfeat_dir: data/feat/feat_vtt_s3d
  text_processor: MSRVTTTextProcessor
  json_path: data/msrvtt/MSRVTT_data.json
  aligner: DSAligner
  num_iso_layer: 12
model:
  model_cls: MMFusionJoint
  mm_encoder_cls: MMBertForJoint
loss:
  loss_cls: T2VContraLoss
fairseq:
  dataset:
    batch_size: 256
  optimization:
    max_epoch: 10
  checkpoint:
    save_dir: runs/task/vtt