test.yaml 300 Bytes
Newer Older
huaerkl's avatar
v1.0  
huaerkl committed
1
2
3
4
5
6
7
8
9
10
11
12
13
# this yaml cannot be run alone: implement a test_${dataset}.yaml
slurm_config: big
task_type: local_predict
dataset:
  split: test
  video_processor: VideoProcessor
  aligner: DSAligner
  bert_name: bert-base-uncased
fairseq:
  dataset:
    batch_size: 256
    valid_subset: test
    num_workers: 2