starcoder icon indicating copy to clipboard operation
starcoder copied to clipboard

Why utilizing the 'question' column of the 'ArmelR/stack-exchange-instruction' dataset for gradient backpropagation?

Open HIT-cwh opened this issue 2 years ago • 1 comments

Thank you for your valuable open-source contribution!

In instruction tuning stage, it seems that only the answer aligned with the instruction participates in the backpropagation process. And this code seems to imply that the question part of the dataset is also involved in backpropagation of the gradient. Will this lead to better training results?

image

HIT-cwh avatar Aug 31 '23 03:08 HIT-cwh

Hi. Thank you for reaching out. In this implementation, we do not mask the part of the sequence which corresponds to the question. You can slightly modify the code to account for that. Both methods work reasonably well (masking and not masking) though masking seems to be the standard practice.

ArmelRandy avatar Sep 01 '23 12:09 ArmelRandy