I get 2.9G model when I train my data?

OpenGVLab / InternImage

[CVPR 2023 Highlight] InternImage: Exploring Large-Scale Vision Foundation Models with Deformable Convolutions

https://arxiv.org/abs/2211.05778

MIT License

2.48k stars 231 forks source link

Open chenzx2 opened 1 year ago

chenzx2 commented 1 year ago

I get 2.9G model when I train my data? It should be 500M.

chenzx2 commented 1 year ago

my config is dino_4scale_internimage_l_1x_coco_0.1x_backbone_lr

monologuesmw commented 1 year ago

I got 3.0G model when I train my data. and got 1.0G onnx model, 500M trt model

chenzx2 commented 1 year ago

I need help, I don't know what is wrong with my code.My model is too big