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.71k stars 1.04k forks source link

How to remove the AGM module when converting the ckpt model #525

Open chenyao96 opened 1 year ago

chenyao96 commented 1 year ago

i convert the ckpt, how to remove the auxiliary training module AGM when converting ckpt to onnx or pth.