├── README.md ├── assets └── pipeline.png ├── data ├── Audiocaps │ ├── test_audiocaps_2label.json │ ├── test_audiocaps_3label.json │ ├── test_audiocaps_subset.json │ ├── train_audiocaps.json │ └── valid_audiocaps.json ├── HA │ ├── 2label_HA.json │ └── 3label_HA.json └── RM │ ├── 2label_RA.json │ ├── 3label_RA.json │ ├── grouped_ratings_2label.json │ └── grouped_ratings_3label.json ├── requirements.txt └── src └── rewardmodel ├── RM_inference_2label_BCE.py └── RM_train_2label_BCE.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/README.md -------------------------------------------------------------------------------- /assets/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/assets/pipeline.png -------------------------------------------------------------------------------- /data/Audiocaps/test_audiocaps_2label.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/data/Audiocaps/test_audiocaps_2label.json -------------------------------------------------------------------------------- /data/Audiocaps/test_audiocaps_3label.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/data/Audiocaps/test_audiocaps_3label.json -------------------------------------------------------------------------------- /data/Audiocaps/test_audiocaps_subset.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/data/Audiocaps/test_audiocaps_subset.json -------------------------------------------------------------------------------- /data/Audiocaps/train_audiocaps.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/data/Audiocaps/train_audiocaps.json -------------------------------------------------------------------------------- /data/Audiocaps/valid_audiocaps.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/data/Audiocaps/valid_audiocaps.json -------------------------------------------------------------------------------- /data/HA/2label_HA.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/data/HA/2label_HA.json -------------------------------------------------------------------------------- /data/HA/3label_HA.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/data/HA/3label_HA.json -------------------------------------------------------------------------------- /data/RM/2label_RA.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/data/RM/2label_RA.json -------------------------------------------------------------------------------- /data/RM/3label_RA.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/data/RM/3label_RA.json -------------------------------------------------------------------------------- /data/RM/grouped_ratings_2label.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/data/RM/grouped_ratings_2label.json -------------------------------------------------------------------------------- /data/RM/grouped_ratings_3label.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/data/RM/grouped_ratings_3label.json -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/rewardmodel/RM_inference_2label_BCE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/src/rewardmodel/RM_inference_2label_BCE.py -------------------------------------------------------------------------------- /src/rewardmodel/RM_train_2label_BCE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hannieliao/Baton/HEAD/src/rewardmodel/RM_train_2label_BCE.py --------------------------------------------------------------------------------