How do we get the number of parameters in the network (SDNN and SNN) ? #236
-
|
Hi, is there a way to print summary like we do in torch to get the number of parameters in the network (SDNN andSNN) ? |
Beta Was this translation helpful? Give feedback.
Answered by
bamsumit
Sep 25, 2023
Replies: 1 comment
-
|
You can use the standard torch way to get the number of parameters: |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
Kristi1217
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
You can use the standard torch way to get the number of parameters:
sum(p.numel() for p in model.parameters())