Update flops_counter.py

open-mmlab · Mar 18, 2021 · e9a8f3c · e9a8f3c
1 parent 6766b17
commit e9a8f3c
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/mmcv/cnn/utils/flops_counter.py b/mmcv/cnn/utils/flops_counter.py
@@ -41,11 +41,11 @@ def get_model_complexity_info(model,
                               flush=False,
                               ost=sys.stdout):
     """Get complexity information of a model.
-    
-    This method can calculate FLOPs
-    and parameter counts of a model with corresponding input shape. It can also
-    print complexity information for each layer in a model.
-    
+
+    This method can calculate FLOPs and parameter counts of a model with
+    corresponding input shape. It can also print complexity information for
+    each layer in a model.
+
     Supported layers are listed as below:
         - Convolutions: ``nn.Conv1d``, ``nn.Conv2d``, ``nn.Conv3d``.
         - Activations: ``nn.ReLU``, ``nn.PReLU``, ``nn.ELU``, ``nn.LeakyReLU``,