-
Notifications
You must be signed in to change notification settings - Fork 102
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
怎么训练自己的数据集? #122
Comments
可以参考我在README里写的fine-tune的guidline. |
感谢作者回复,但是我现在出了点问题,微调开始后很快结束,看起来并没有开始训练,我不知道是不是我的训练集或是config,sh文件配置错误,我刚开始学习,不太懂,请见谅,希望能够得到大家的解答 |
可以再看下guideline -- “Attention: the epoch of training continues from the epochs the weights file name indicates (e.g., 244 in BiRefNet-general-epoch_244.pth), instead of 1. ”. |
|
什么型号的GPU? 想fine-tune之前我给的标准版的BiRefNet的话, 得需要>36GB的卡, 多卡并行的话甚至要>41GB. |
No description provided.
The text was updated successfully, but these errors were encountered: