RangiLyu / nanodet

NanoDet-Plusāš”Super fast and lightweight anchor-free object detection model. šŸ”„Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphonešŸ”„
Apache License 2.0
5.78k stars 1.04k forks source link

AMP training bug #494

Open lonelyaustralis opened 1 year ago

lonelyaustralis commented 1 year ago

If you want to use AMP training, in addition to setting the precision to 16, you also need to add amp_backend="native" to the trainer, otherwise it will not have any effect