How to get NUMBER OF PARAMETERS used?

qfgaohao / pytorch-ssd

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.

https://medium.com/@smallfishbigsea/understand-ssd-and-implement-your-own-caa3232cd6ad

MIT License

1.39k stars 529 forks source link

How to get NUMBER OF PARAMETERS used? #155

Open Lucky2593 opened 3 years ago

Lucky2593 commented 3 years ago

Can someone please help in telling me how can I get the number of parameters used in the network.

I am training "mobileNetv1-SSDLite" as it started training from scratch(Training base network also).

Command used to train "MobileNetV1-SSDLite"

python train_ssd.py --datasets D:/DataSets/VOCdevkit/VOC2007 D:/DataSets/VOCdevkit/VOC2012 --validation_dataset D:/DataSets/VOCdevkit/VOC2007 --net mb1-ssd-lite --batch_size 24 --num_epochs 200 --scheduler cosine --lr 0.01 --t_max 200