Closed Bingo-1996 closed 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.
Hi~ How many parameters does CenterSnap contain?