diff --git a/configs/picodet/README.md b/configs/picodet/README.md index 0ef0d6734b47b465f394e42deefb07a909e5a55c..9c145b0de0a6807fef706f09d4b6adabe7c4533a 100644 --- a/configs/picodet/README.md +++ b/configs/picodet/README.md @@ -308,6 +308,21 @@ Please use `PicoDet-LCNet` model, which has fewer `transpose` operators. +
+How to count model parameters. + +You can insert below code at [here](https://github.com/PaddlePaddle/PaddleDetection/blob/develop/ppdet/engine/trainer.py#L141) to count learnable parameters. + +```python +params = sum([ + p.numel() for n, p in self.model. named_parameters() + if all([x not in n for x in ['_mean', '_variance']]) +]) # exclude BatchNorm running status +print('params: ', params) +``` + +
+ ## Cite PP-PicoDet If you use PicoDet in your research, please cite our work by using the following BibTeX entry: ```