ViTPose About model size

Hi，I used the pre-trained model you provided for fine-tuning. Performance and speed is competitive.But the size of the model is about three times larger than you.For example, the size of my vitpose-b is 1.xGB, but yours is 343MB. How can i get a same size model?

Nov 09 '22 02:11 glamourzc

Thanks for your interest. I think the size you talk about is the size of the checkpoint, right? The default checkpoint saved contains not only the model's weights but also the optimizer's state. Maybe this is the reason for the larger size of the saved checkpoint.

Nov 09 '22 11:11 Annbless

Yaeh. My onnx model and trt-fp32 model are the same size as the pth you provided.Thx.

Nov 11 '22 03:11 glamourzc

@glamourzc hi, could you provide the onnx model. I try to convert the vitae-pose B* model but somehow it doesn't work

Nov 28 '22 06:11 SonNguyen2510

It seems there are no further questions. I will close this issue temporarily. If you have some further questions, please feel free to re-open it.

Feb 13 '23 03:02 Annbless