From fca30a74915971181da5f98ce107a474b0b7d47b Mon Sep 17 00:00:00 2001
From: Rajarshi Banerjee <rajarshiban13@gmail.com>
Date: Mon, 8 Aug 2022 20:46:55 +0530
Subject: [PATCH] Update readme.md

---
 readme.md | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/readme.md b/readme.md
index 75f13e0..ecec0df 100644
--- 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}
- ```
\ No newline at end of file
+ ```