mathvista.yaml 604 Bytes
Newer Older
Baber's avatar
Baber committed
1
dataset_path: AI4Math/MathVista
Baber's avatar
Baber committed
2
task: mathvista
Baber's avatar
Baber committed
3
test_split: testmini
Baber's avatar
Baber committed
4
5
output_type: "generate_until"
#process_docs: !function utils.process_docs
Baber's avatar
Baber committed
6
7
doc_to_image:
  - decoded_image
Baber's avatar
Baber committed
8
doc_to_text: "<image>{{query}}"
Baber's avatar
Baber committed
9
10
11
#doc_to_choice: '{{ ["A", "B", "C", "D", "E", "F"][:choices.length] }}'
doc_to_target: answer
process_results: !function utils.process_results
Baber's avatar
Baber committed
12
13
14
15
16
generation_kwargs:
  until:
    - "<|endoftext|>"
  temperature: 0.0
  do_sample: false
Baber's avatar
Baber committed
17
  max_gen_toks: 1024
Baber's avatar
Baber committed
18
19
20
21
22
23
24
25
metric_list:
  - metric: acc
    aggregation: mean
    higher_is_better: true
metadata:
  version: 1.0
dataset_kwargs:
  trust_remote_code: true