HiTZ/lmloss-opt-rm-1.3b

Model

LM Loss OPT RM

0

7 commits

1 linked in READMEs

updated Apr 7, 2023

See the code

README

LM Loss OPT RM

This is a fine tuned OPT 1.3b model for reward modelling. The finetuning has been done on top of the full SLF5K dataset following the method presented in the paper Training Language Models with Language Feedback at Scale. The main results can be seen in the following table:

Model# ParamsValidation Accuracy (in %)
OPT LM Loss13B73.4 +/- 1.9
OPT LM Loss1.3B69.6 +/- 2.0
OPT RM Loss13B71.8 +/- 2.0

If using this model, please cite the following paper:

@article{scheurer2023training,
  title={Training Language Models with Language Feedback at Scale},
  author={Scheurer, J{\'e}r{\'e}my and Campos, Jon Ander and Korbak, Tomasz and Chan, Jun Shern and Chen, Angelica and Cho, Kyunghyun and Perez, Ethan},
  journal={arXiv preprint arXiv:2303.16755},
  year={2023}
}
endpoints_compatible
pytorch
text-generation
text-generation-inference
transformers

Contributors

ixa-ehu

4 commits

jcampos

3 commits

HiTZ/lmloss-opt-rm-1.3b

Model

LM Loss OPT RM

0

7 commits

1 linked in READMEs

updated Apr 7, 2023

See the code

README

LM Loss OPT RM

This is a fine tuned OPT 1.3b model for reward modelling. The finetuning has been done on top of the full SLF5K dataset following the method presented in the paper Training Language Models with Language Feedback at Scale. The main results can be seen in the following table:

Model# ParamsValidation Accuracy (in %)
OPT LM Loss13B73.4 +/- 1.9
OPT LM Loss1.3B69.6 +/- 2.0
OPT RM Loss13B71.8 +/- 2.0

If using this model, please cite the following paper:

@article{scheurer2023training,
  title={Training Language Models with Language Feedback at Scale},
  author={Scheurer, J{\'e}r{\'e}my and Campos, Jon Ander and Korbak, Tomasz and Chan, Jun Shern and Chen, Angelica and Cho, Kyunghyun and Perez, Ethan},
  journal={arXiv preprint arXiv:2303.16755},
  year={2023}
}
endpoints_compatible
pytorch
text-generation
text-generation-inference
transformers

Contributors

ixa-ehu

4 commits

jcampos

3 commits