the model.pth file is too large

tianzhi0549 / FCOS

FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)

https://arxiv.org/abs/1904.01355

Other

3.28k stars 630 forks source link

Closed dreamhighchina closed 5 years ago

dreamhighchina commented 5 years ago

when i have trained fcos_R_101_FPN_2x,the model.pth is 399M which is huge different from your 200M.my class is 7.could tell me the possible reason???

milliema commented 5 years ago

I found the real reason, after removing the scheduler, optimizer, and iteration, the model size will be decreased to the same as author provided.

dreamhighchina commented 5 years ago

how to remove these？

milliema commented 5 years ago

in tools/remove_solver_states.py

dreamhighchina commented 5 years ago

in tools/remove_solver_states.py

thank you for your reply，how can i communicate with you to talk about this model about other questions？