PRIMA / code /configs /evaluate.yaml
kanguyen-vn's picture
Include clean PRIMA code and evaluator; document metrics, initialization, and MIT contributions
91c1e9e verified
Raw History Blame Contribute Delete
237 Bytes
merged_ckpt_path: PLAN-Lab/PRIMA
dataset_dir: /path/to/datasets
m4seg_annpath: /path/to/datasets/m4seg/annotations/m4seg_test.json.gz
precision: bf16
val_batch_size: 1
max_new_tokens: 1024
workers: 0
output_save_path: outputs/evaluation