The loss function of reward model.
Author: huzechuanCreated Jan 31, 2023Updated Feb 12, 2023
Hi, I am confused that the loss function of ChatGPT's reward model takes as input the difference of two responses and then passes a sigmoid function. However, the loss function in this repo only takes one response as input and uses the ranking score as a label to calculate the CE loss. Is there an advantage to this?
Source: lucidrains/PaLM-rlhf-pytorch