关于混合精度训练. - Githubissues

tinyvision / DAMO-YOLO

DAMO-YOLO: a fast and accurate object detection method with some new techs, including NAS backbones, efficient RepGFPN, ZeroHead, AlignedOTA, and distillation enhancement.

Apache License 2.0

3.75k stars 470 forks source link

关于混合精度训练. #74

Closed z1069614715 closed 1 year ago

z1069614715 commented 1 year ago

Before Asking

[X] I have read the README carefully. 我已经仔细阅读了README上的操作指引。
[X] I want to train my custom dataset, and I have read the tutorials for finetune on your data carefully and organize my dataset correctly; 我想训练自定义数据集，我已经仔细阅读了训练自定义数据的教程，以及按照正确的目录结构存放数据集。
[X] I have pulled the latest code of main branch to run again and the problem still existed. 我已经拉取了主分支上最新的代码，重新运行之后，问题仍不能解决。

Search before asking

[X] I have searched the DAMO-YOLO issues and found no similar questions.

Question

DAMO-YOLO的作者你好，我想问问你们的项目不使用混合精度训练的原因？是因为混合精度有可能带来loss为nan的问题吗？

Additional

No response

cwhgn commented 1 year ago

感谢建议，的确目前的版本是没有混合精度训练的。这是个很好的建议，我们后续会把它加进来。

z1069614715 commented 1 year ago

感谢回复，期待你们的更新.