Update readme.md

GSAUC3 · Aug 8, 2022 · fca30a7 · fca30a7
1 parent 5d88be0
commit fca30a7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/readme.md b/readme.md
@@ -130,8 +130,9 @@ Total Non-Trainable parameters 0
 Total Trainable parameters 61,100,840
 ```
 
+### Calculating the number of specific layer, or layer frequencies
 ```python
-from pytorchsummary import get_num_layers(model)
+from pytorchsummary import get_num_layers
 print(get_num_layers(m)) # alexnet model 
 ```
 Output:
@@ -142,4 +143,4 @@ Output:
  'AdaptiveAvgPool2d': 1,
  'Dropout': 2,
  'Linear': 3}
- ```
+ ```