_template_yaml 800 Bytes
Newer Older
lintangsutawika's avatar
lintangsutawika committed
1
dataset_path: MMMU/MMMU
Ashvin Nihalani's avatar
Ashvin Nihalani committed
2
3
4
5
6
7
8
task: "mmmu_val"
validation_split: validation
output_type: generate_until
doc_to_visual: !function utils.mmmu_doc_to_visual
doc_to_text: !function utils.mmmu_doc_to_text
doc_to_target: "answer"
# The return value of process_results will be used by metrics
lintangsutawika's avatar
lintangsutawika committed
9
# process_results: !function utils.mmmu_process_results
Ashvin Nihalani's avatar
Ashvin Nihalani committed
10
11
12
13
14
15
16
17
18
19
20
# Note that the metric name can be either a registed metric function (such as the case for GQA) or a key name returned by process_results
generation_kwargs:
  until:
    - "<|endoftext|>"
  temperature: 1.0
  do_sample: false
  max_gen_toks: 512
  top_p: 0.94
  repetition_penalty: 1.0
  image_aspect_ratio: original
metric_list:
lintangsutawika's avatar
lintangsutawika committed
21
  - metric: exact_match
lintangsutawika's avatar
lintangsutawika committed
22
23
24
  # - metric: mmmu_acc
  #   aggregation: !function utils.mmmu_aggregate_results
  #   higher_is_better: true