search

zubair-irshad / CenterSnap

Pytorch code for ICRA'22 paper: "Single-Shot Multi-Object 3D Shape Reconstruction and Categorical 6D Pose and Size Estimation"
https://zubair-irshad.github.io/projects/CenterSnap.html
286 stars 47 forks source link

Parameters count #4

Closed Bingo-1996 closed 2 years ago

Bingo-1996 commented 2 years ago

Hi~ How many parameters does CenterSnap contain?

zubair-irshad commented 2 years ago

The total number of trainable parameters for CenterSnap model is around 5M.

You can print them using pytorch_total_params = sum(p.numel() for p in model.model.parameters() if p.requires_grad) in net_train.py or use pytorch_summary.