args:
{'add_timestamp': False,
'hf_entity': 'jacobmorrison',
'hf_repo_id': 'rejection_sampling_6328',
'hf_repo_id_scores': 'scores_6328',
'input_filename': '/output/shards/6328/3.jsonl',
'max_forward_batch_size': 64,
'mode': 'judgement',
'model_names_or_paths': ['/reward_model'],
'num_completions':… See the full description on the dataset page:
https://huggingface.co/datasets/jacobmorrison/scores_6328.