ultralytics-thop 2.0.17__py3-none-any.whl → 2.0.18__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (9) hide show
  1. thop/__init__.py +2 -2
  2. thop/fx_profile.py +2 -2
  3. thop/profile.py +1 -1
  4. {ultralytics_thop-2.0.17.dist-info → ultralytics_thop-2.0.18.dist-info}/METADATA +1 -1
  5. ultralytics_thop-2.0.18.dist-info/RECORD +13 -0
  6. ultralytics_thop-2.0.17.dist-info/RECORD +0 -13
  7. {ultralytics_thop-2.0.17.dist-info → ultralytics_thop-2.0.18.dist-info}/WHEEL +0 -0
  8. {ultralytics_thop-2.0.17.dist-info → ultralytics_thop-2.0.18.dist-info}/licenses/LICENSE +0 -0
  9. {ultralytics_thop-2.0.17.dist-info → ultralytics_thop-2.0.18.dist-info}/top_level.txt +0 -0
thop/__init__.py CHANGED
@@ -1,6 +1,6 @@
1
1
  # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
2
2
 
3
- __version__ = "2.0.17"
3
+ __version__ = "2.0.18"
4
4
 
5
5
  import torch
6
6
 
@@ -9,4 +9,4 @@ from .utils import clever_format
9
9
 
10
10
  default_dtype = torch.float64
11
11
 
12
- __all__ = ["profile", "profile_origin", "clever_format", "default_dtype"]
12
+ __all__ = ["clever_format", "default_dtype", "profile", "profile_origin"]
thop/fx_profile.py CHANGED
@@ -49,11 +49,11 @@ def count_fn_conv2d(input_shapes, output_shapes, *args, **kwargs):
49
49
  """Calculates total operations (FLOPs) for a 2D conv layer based on input and output shapes using
50
50
  `calculate_conv`.
51
51
  """
52
- inputs, weight, bias, stride, padding, dilation, groups = args
52
+ _inputs, _weight, _bias, _stride, _padding, _dilation, groups = args
53
53
  if len(input_shapes) == 2:
54
54
  x_shape, k_shape = input_shapes
55
55
  elif len(input_shapes) == 3:
56
- x_shape, k_shape, b_shape = input_shapes
56
+ x_shape, k_shape, _b_shape = input_shapes
57
57
  out_shape = output_shapes[0]
58
58
 
59
59
  kernel_parameters = k_shape[2:].numel()
thop/profile.py CHANGED
@@ -94,7 +94,7 @@ def profile_origin(model, inputs, custom_ops=None, verbose=True, report_missing=
94
94
 
95
95
  if hasattr(m, "total_ops") or hasattr(m, "total_params"):
96
96
  logging.warning(
97
- f"Either .total_ops or .total_params is already defined in {str(m)}. "
97
+ f"Either .total_ops or .total_params is already defined in {m!s}. "
98
98
  "Be careful, it might change your code's behavior."
99
99
  )
100
100
 
{ultralytics_thop-2.0.17.dist-info → ultralytics_thop-2.0.18.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: ultralytics-thop
3
- Version: 2.0.17
3
+ Version: 2.0.18
4
4
  Summary: Ultralytics THOP package for fast computation of PyTorch model FLOPs and parameters.
5
5
  Author-email: Ligeng Zhu <ligeng.zhu+github@gmail.com>
6
6
  Maintainer-email: Ultralytics <hello@ultralytics.com>
ultralytics_thop-2.0.18.dist-info/RECORD ADDED
@@ -0,0 +1,13 @@
1
+ thop/__init__.py,sha256=EBsQOr5rVWk_i2g7NxrycotRXfKmpgcUu1tDBMtJ0Jw,293
2
+ thop/fx_profile.py,sha256=uQETQqeLlOmx_AVunvdZ27xX13a8qVHk7zmjQrd-Yec,8411
3
+ thop/profile.py,sha256=D8OpFcM66YKthAkxnEgdOquuUdsXKnyKbxnOJE9GBM4,8354
4
+ thop/rnn_hooks.py,sha256=JKZ2eSCvIKvhvCDqM4oWPZjmBkdyJ4R2Q7XSn63lsX0,6503
5
+ thop/utils.py,sha256=IwFJQ1v-SLyhm-313Li535R6fhtomkm8Fem1Kfe6G_U,1484
6
+ thop/vision/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6DXppv1-QUM,70
7
+ thop/vision/basic_hooks.py,sha256=tvr8P4tRJpP9Qyk-tnuXnBPGaMGBj6qlDpGI-uJh5mE,5466
8
+ thop/vision/calc_func.py,sha256=2-LcoFg8ODy8EMb-xEsgLVPigLTki3dXnAg58dxVq80,4530
9
+ ultralytics_thop-2.0.18.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
10
+ ultralytics_thop-2.0.18.dist-info/METADATA,sha256=bvyzh4AhrPj8Q8yrbPSNBe4nDGwTzZ9HSlgKDGNZvzc,14942
11
+ ultralytics_thop-2.0.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
12
+ ultralytics_thop-2.0.18.dist-info/top_level.txt,sha256=HQ7D0gSvDJ31CNR-f0EuXNVve05RYBmwyIkHQKiEhU8,5
13
+ ultralytics_thop-2.0.18.dist-info/RECORD,,
ultralytics_thop-2.0.17.dist-info/RECORD DELETED
@@ -1,13 +0,0 @@
1
- thop/__init__.py,sha256=jUflh8SMGvE0AqZliYPEX-DJ0ynmVJVMIHKB9IU6BlQ,293
2
- thop/fx_profile.py,sha256=mdomj44BIc85XyDabnYRjlJI_p3BxAMls0P8GVS2WYU,8404
3
- thop/profile.py,sha256=c5aH5zfL5e68OJtNL3OO45nkH51PygrE325wMQ4H2pE,8357
4
- thop/rnn_hooks.py,sha256=JKZ2eSCvIKvhvCDqM4oWPZjmBkdyJ4R2Q7XSn63lsX0,6503
5
- thop/utils.py,sha256=IwFJQ1v-SLyhm-313Li535R6fhtomkm8Fem1Kfe6G_U,1484
6
- thop/vision/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6DXppv1-QUM,70
7
- thop/vision/basic_hooks.py,sha256=tvr8P4tRJpP9Qyk-tnuXnBPGaMGBj6qlDpGI-uJh5mE,5466
8
- thop/vision/calc_func.py,sha256=2-LcoFg8ODy8EMb-xEsgLVPigLTki3dXnAg58dxVq80,4530
9
- ultralytics_thop-2.0.17.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
10
- ultralytics_thop-2.0.17.dist-info/METADATA,sha256=vHUOSsXrncSfOCM2J7dQLcayh-fHVDwZZVV40meK7yE,14942
11
- ultralytics_thop-2.0.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
12
- ultralytics_thop-2.0.17.dist-info/top_level.txt,sha256=HQ7D0gSvDJ31CNR-f0EuXNVve05RYBmwyIkHQKiEhU8,5
13
- ultralytics_thop-2.0.17.dist-info/RECORD,,