mathvista.yaml 599 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
6
output_type: "generate_until"
#process_docs: !function utils.process_docs
doc_to_image: decoded_image
Baber's avatar
Baber committed
7
8
9
10
doc_to_text: "<image> {{query}}"
#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
11
12
13
14
15
16
generation_kwargs:
  until:
    - "<|endoftext|>"
  temperature: 0.0
  do_sample: false
  max_gen_toks: 64
Baber's avatar
Baber committed
17
18
19
20
21
22
23
24
metric_list:
  - metric: acc
    aggregation: mean
    higher_is_better: true
metadata:
  version: 1.0
dataset_kwargs:
  trust_remote_code: true