How can i custom training loop, sampling data and loss function in SFTTrainer #1433
Answered
by
younesbelkada
nguyenhoanganh2002
asked this question in
Q&A
-
|
I want to change training loop, sample data, loss func. Same in this repo https://github.com/nju-websoft/SCKD/tree/main. |
Beta Was this translation helpful? Give feedback.
Answered by
younesbelkada
Apr 8, 2024
Replies: 1 comment
-
|
hi @nguyenhoanganh2002 |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
nguyenhoanganh2002
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