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.

Sign up for GitHub

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

Jump to bottom

是否支持win系统 voc数据集训练？ #47

Closed

Gaoang112 opened this issue Oct 25, 2022 · 1 comment

Gaoang112 commented Oct 25, 2022

在win系统下自定义 voc数据集训练，总是内存溢出，将bs调到1依然不行！

Collaborator

nemonameless commented Oct 26, 2022 •

edited

Loading

支持的。请先参考 #43 ，voc数据集也一样。
训练命令后加--amp 开启混合精度训练可以降低显存占用，建议训yolov7_tiny模型，否则bs过小检测精度会极低，原版yolov7总bs是8x32=256。

nemonameless closed this as completed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment