Skip to content
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

Open
Whiteeeeeeeeeeeeeeeeee opened this issue Nov 13, 2024 · 5 comments
Open

怎么训练自己的数据集? #122

Whiteeeeeeeeeeeeeeeeee opened this issue Nov 13, 2024 · 5 comments

Comments

@Whiteeeeeeeeeeeeeeeeee
Copy link

No description provided.

@ZhengPeng7
Copy link
Owner

可以参考我在README里写的fine-tune的guidline.

@Whiteeeeeeeeeeeeeeeeee
Copy link
Author

可以参考我在README里写的fine-tune的guidline.

感谢作者回复,但是我现在出了点问题,微调开始后很快结束,看起来并没有开始训练,我不知道是不是我的训练集或是config,sh文件配置错误,我刚开始学习,不太懂,请见谅,希望能够得到大家的解答
image
image
image

@ZhengPeng7
Copy link
Owner

可以再看下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. ”.

@Whiteeeeeeeeeeeeeeeeee
Copy link
Author

可以再看下guideline -- “注意:训练纪元从权重文件名指示的纪元开始(例如,BiRefNet-general-epoch_244.pth 中的 244),而不是 1。
哦哦我知道了epoch的问题,谢谢作者耐心回答,但是我的显卡一直溢出QAQ,换了几个pth还是不行,你的模型很棒但是我用不了了呜呜,祝好

@ZhengPeng7
Copy link
Owner

什么型号的GPU? 想fine-tune之前我给的标准版的BiRefNet的话, 得需要>36GB的卡, 多卡并行的话甚至要>41GB.
BiRefNet_lite的话4090这种24GB的就可以跑.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants