search

Deci-AI / super-gradients

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.
https://www.supergradients.com
Apache License 2.0
4.54k stars 496 forks source link

Error message: "mixed precision is not available for CPU" even with GPU setting on. #1547

Closed tztechno closed 11 months ago

tztechno commented 11 months ago

💡 Your Question

Recently, in YOLO-NAS custom training, when setting "mixed_precision": True along with GPU setting on, AssertionError occurred with this message: "mixed precision is not available for CPU." and program stopped even with GPU setting on. I know this error was not seen before. Is there anyone encountering the same phenomenon?

Versions

No response

BloodAxe commented 11 months ago

Thanks, this has been fixed in 3.3.1 release: https://github.com/Deci-AI/super-gradients/releases/tag/3.3.1

tztechno commented 11 months ago

I confirmed that it can be executed using GPU/CPU. Thank you.