ultralytics-thop 2.0.17__tar.gz → 2.0.18__tar.gz

{ultralytics_thop-2.0.17 → ultralytics_thop-2.0.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics-thop
-Version: 2.0.17
+Version: 2.0.18
 Summary: Ultralytics THOP package for fast computation of PyTorch model FLOPs and parameters.
 Author-email: Ligeng Zhu <ligeng.zhu+github@gmail.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>

{ultralytics_thop-2.0.17 → ultralytics_thop-2.0.18}/tests/test_conv2d.py

@@ -21,7 +21,7 @@ class TestUtils:
 
         _, _, oh, ow = out.shape
         print(f"Conv2d: in={ih}x{iw}, kernel={kh}x{kw}, stride={s}, padding={p}, out={oh}x{ow}")
-        macs, params = profile(net, inputs=(data,))
+        macs, _params = profile(net, inputs=(data,))
         assert macs == 810000, f"{macs} v.s. 810000"
 
     def test_conv2d(self):
@@ -36,7 +36,7 @@ class TestUtils:
 
         _, _, oh, ow = out.shape
         print(f"Conv2d: in={ih}x{iw}, kernel={kh}x{kw}, stride={s}, padding={p}, out={oh}x{ow}")
-        macs, params = profile(net, inputs=(data,))
+        macs, _params = profile(net, inputs=(data,))
         assert macs == 810000, f"{macs} v.s. 810000"
 
     def test_conv2d_random(self):
@@ -54,7 +54,7 @@ class TestUtils:
 
             _, _, oh, ow = out.shape
             print(f"Conv2d: in={ih}x{iw}, kernel={kh}x{kw}, stride={s}, padding={p}, out={oh}x{ow}")
-            macs, params = profile(net, inputs=(data,))
+            macs, _params = profile(net, inputs=(data,))
             assert macs == n * out_c * oh * ow // g * in_c * kh * kw, (
                 f"{macs} v.s. {n * out_c * oh * ow // g * in_c * kh * kw}"
             )

{ultralytics_thop-2.0.17 → ultralytics_thop-2.0.18}/thop/__init__.py

@@ -1,6 +1,6 @@
 # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
 
-__version__ = "2.0.17"
+__version__ = "2.0.18"
 
 import torch
 
@@ -9,4 +9,4 @@ from .utils import clever_format
 
 default_dtype = torch.float64
 
-__all__ = ["profile", "profile_origin", "clever_format", "default_dtype"]
+__all__ = ["clever_format", "default_dtype", "profile", "profile_origin"]

{ultralytics_thop-2.0.17 → ultralytics_thop-2.0.18}/thop/fx_profile.py

@@ -49,11 +49,11 @@ 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`.
     """
-    inputs, weight, bias, stride, padding, dilation, groups = args
+    _inputs, _weight, _bias, _stride, _padding, _dilation, groups = args
     if len(input_shapes) == 2:
         x_shape, k_shape = input_shapes
     elif len(input_shapes) == 3:
-        x_shape, k_shape, b_shape = input_shapes
+        x_shape, k_shape, _b_shape = input_shapes
     out_shape = output_shapes[0]
 
     kernel_parameters = k_shape[2:].numel()

{ultralytics_thop-2.0.17 → ultralytics_thop-2.0.18}/thop/profile.py

@@ -94,7 +94,7 @@ def profile_origin(model, inputs, custom_ops=None, verbose=True, report_missing=
 
         if hasattr(m, "total_ops") or hasattr(m, "total_params"):
             logging.warning(
-                f"Either .total_ops or .total_params is already defined in {str(m)}. "
+                f"Either .total_ops or .total_params is already defined in {m!s}. "
                 "Be careful, it might change your code's behavior."
             )
 

{ultralytics_thop-2.0.17 → ultralytics_thop-2.0.18}/ultralytics_thop.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics-thop
-Version: 2.0.17
+Version: 2.0.18
 Summary: Ultralytics THOP package for fast computation of PyTorch model FLOPs and parameters.
 Author-email: Ligeng Zhu <ligeng.zhu+github@gmail.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>