search

project-baize / baize-chatbot

Let ChatGPT teach your own chatbot in hours with a single GPU!
https://arxiv.org/abs/2304.01196
GNU General Public License v3.0
3.16k stars 282 forks source link

where is the loss code? #49

Open royshan opened 1 year ago

royshan commented 1 year ago

First, I'd like to thank you very much for this inspiring work!

In your paper, you mentioned "For Baize v1.5, following Vicuna (Chiang et al., 2023), we only compute loss for AI’s responses in the dialogue transcript." I am curious about your implementation and may customize it. I searched the repo but couldn't find it. Could you give me a pointer for that?

albertwy commented 1 year ago

Maybe the version of this code is too old