How can i custom training loop, sampling data and loss function in SFTTrainer #1433
|
I want to change training loop, sample data, loss func. Same in this repo https://github.com/nju-websoft/SCKD/tree/main. |
Answered by
younesbelkada
Apr 8, 2024
Replies: 1 comment
|
hi @nguyenhoanganh2002 |
0 replies
Answer selected by
anhnh2002
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
hi @nguyenhoanganh2002
Thanks for the issue! I think the easiest is to subclass
STTrainerto create your own trainer and overwrite the desired methods