PyTorch 2 · Autograd and models
Count a model's parameters
Count trainable parameters.
sum(p.numel() for p in model.parameters() if p.requires_grad)
Official PyTorch 2 documentation
PyTorch 2 · Autograd and models
Count trainable parameters.
sum(p.numel() for p in model.parameters() if p.requires_grad)
Official PyTorch 2 documentation
Was this page helpful?