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

cpu训练正常，gpu训练出现loss=nan #634

Open

DragMou1204 opened this issue Jan 13, 2024 · 1 comment

DragMou1204 commented Jan 13, 2024

我训练的是自制的数据集，1000×500的tif图片
只更改了config.py中的图片和标签的dateset.txt和clsss.names

DragMou1204 changed the title ~~训练生成的文件名loss显示nan，且推理没有结果~~ cpu训练正常，gpu训练出现loss=nan

Author

DragMou1204 commented Jan 13, 2024

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