search

Megvii-BaseDetection / YOLOX

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/
Apache License 2.0
9.44k stars 2.21k forks source link

yolox-s.pth 2 yolox-s.onnx question？ #495

Closed sunmooncode closed 3 years ago

sunmooncode commented 3 years ago

yolox-s.pth weight size 71.4M yolox-s.onnx weight size 35.7M Why is there so much difference？

Joker316701882 commented 3 years ago

@sunmooncode yolox-s.pth contains not only model parameters, but also the information in the optimizer. For example, the momentum in SGD for each parameter. However, the yolox-s.onnx only contains model's parameters.

sunmooncode commented 3 years ago

@Joker316701882 thanks