Auto-format by https://ultralytics.com/actions

ultralytics · Jun 30, 2024 · 04ca8ce · 04ca8ce
1 parent c8ac820
commit 04ca8ce
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 2 deletions.
diff --git a/benchmark/evaluate_famous_models.py b/benchmark/evaluate_famous_models.py
@@ -1,7 +1,8 @@
 import torch
-from thop.profile import profile
 from torchvision import models
 
+from thop.profile import profile
+
 model_names = sorted(
     name
     for name in models.__dict__

diff --git a/benchmark/evaluate_rnn_models.py b/benchmark/evaluate_rnn_models.py
@@ -1,5 +1,6 @@
 import torch
 import torch.nn as nn
+
 from thop.profile import profile
 
 input_size = 160

diff --git a/tests/test_conv2d.py b/tests/test_conv2d.py
@@ -1,5 +1,6 @@
 import torch
 import torch.nn as nn
+
 from thop import profile
 
 

diff --git a/tests/test_matmul.py b/tests/test_matmul.py
@@ -1,5 +1,6 @@
 import torch
 import torch.nn as nn
+
 from thop import profile
 
 

diff --git a/tests/test_relu.py b/tests/test_relu.py
@@ -1,5 +1,6 @@
 import torch
 import torch.nn as nn
+
 from thop import profile
 
 

diff --git a/thop/fx_profile.py b/thop/fx_profile.py
@@ -43,7 +43,9 @@ def count_fn_linear(input_shapes, output_shapes, *args, **kwargs):
 
 
 def count_fn_conv2d(input_shapes, output_shapes, *args, **kwargs):
-    """Calculates total operations (FLOPs) for a 2D conv layer based on input and output shapes using `calculate_conv`."""
+    """Calculates total operations (FLOPs) for a 2D conv layer based on input and output shapes using
+    `calculate_conv`.
+    """
     inputs, weight, bias, stride, padding, dilation, groups = args
     if len(input_shapes) == 2:
         x_shape, k_shape = input_shapes