ultralytics-thop 2.0.6__py3-none-any.whl → 2.0.8__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.
thop/__init__.py CHANGED
@@ -1,4 +1,5 @@
1
- __version__ = "2.0.6"
1
+ __version__ = "2.0.8"
2
+
2
3
 
3
4
  import torch
4
5
 
thop/profile.py CHANGED
@@ -69,8 +69,8 @@ def profile_origin(model, inputs, custom_ops=None, verbose=True, report_missing=
69
69
 
70
70
  if hasattr(m, "total_ops") or hasattr(m, "total_params"):
71
71
  logging.warning(
72
- "Either .total_ops or .total_params is already defined in %s. "
73
- "Be careful, it might change your code's behavior." % str(m)
72
+ f"Either .total_ops or .total_params is already defined in {str(m)}. "
73
+ "Be careful, it might change your code's behavior."
74
74
  )
75
75
 
76
76
  m.register_buffer("total_ops", torch.zeros(1, dtype=default_dtype))
@@ -44,7 +44,7 @@ def count_convNd_ver2(m: _ConvNd, x, y: torch.Tensor):
44
44
  x = x[0]
45
45
 
46
46
  # N x H x W (exclude Cout)
47
- output_size = torch.zeros((y.size()[:1] + y.size()[2:])).numel()
47
+ output_size = torch.zeros(y.size()[:1] + y.size()[2:]).numel()
48
48
  # # Cout x Cin x Kw x Kh
49
49
  # kernel_ops = m.weight.nelement()
50
50
  # if m.bias is not None:
@@ -1,13 +1,15 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ultralytics-thop
3
- Version: 2.0.6
3
+ Version: 2.0.8
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
- Maintainer: Glenn Jocher
6
+ Maintainer-email: Ultralytics <hello@ultralytics.com>
7
7
  License: AGPL-3.0
8
+ Project-URL: Homepage, https://ultralytics.com
9
+ Project-URL: Source, https://github.com/ultralytics/thop
10
+ Project-URL: Documentation, https://docs.ultralytics.com
8
11
  Project-URL: Bug Reports, https://github.com/ultralytics/thop/issues
9
- Project-URL: Funding, https://ultralytics.com
10
- Project-URL: Source, https://github.com/ultralytics/thop/
12
+ Project-URL: Changelog, https://github.com/ultralytics/thop/releases
11
13
  Keywords: FLOPs,PyTorch,Model Analysis
12
14
  Classifier: Development Status :: 4 - Beta
13
15
  Classifier: Intended Audience :: Developers
@@ -176,7 +178,7 @@ THOP is licensed under the AGPL-3.0 License. For more information, see the [LICE
176
178
 
177
179
  ## 📮 Contact
178
180
 
179
- For bugs or feature requests, please open an issue on [GitHub Issues](https://github.com/ultralytics/thop/issues). Join our community on [Discord](https://ultralytics.com/discord) for discussions and support.
181
+ For bugs or feature requests, please open an issue on [GitHub Issues](https://github.com/ultralytics/thop/pulls). Join our community on [Discord](https://discord.com/invite/ultralytics) for discussions and support.
180
182
 
181
183
  <br>
182
184
  <div align="center">
@@ -0,0 +1,13 @@
1
+ thop/__init__.py,sha256=AumnJ91nRmWDiM7Z57GWFrMOZpPzI3g82PP_S-UXDHo,147
2
+ thop/fx_profile.py,sha256=u1eNvsRBTl9u0jJ6ymgrnrJP1VSLpAZkaEvm4D2mrI4,8172
3
+ thop/profile.py,sha256=xIetT1iovk7t9iLvFajF8FqH2z_anItekOi93oeW9A4,7909
4
+ thop/rnn_hooks.py,sha256=IwxLvVvbeo5skThdg3BKXm08_0GsM8FI2QM_alQcCB8,6432
5
+ thop/utils.py,sha256=SP5QWanMqVRieDoYGQN-7lftZmYXohn2E_m3cP17tS4,1413
6
+ thop/vision/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
7
+ thop/vision/basic_hooks.py,sha256=mYMNGI8vfP7OafqBirkt4G_5Wa5upfsTsrWj9Fiu0Ao,4637
8
+ thop/vision/calc_func.py,sha256=5YbWp48XRbJYzSRkAFMr5xlM4o5BbjTcU6rHSKPkD6I,4291
9
+ ultralytics_thop-2.0.8.dist-info/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
10
+ ultralytics_thop-2.0.8.dist-info/METADATA,sha256=wM_0Sf0wluoiqSNWRDrrGggqYx3SEJaj1zw61SGucEs,9284
11
+ ultralytics_thop-2.0.8.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
12
+ ultralytics_thop-2.0.8.dist-info/top_level.txt,sha256=HQ7D0gSvDJ31CNR-f0EuXNVve05RYBmwyIkHQKiEhU8,5
13
+ ultralytics_thop-2.0.8.dist-info/RECORD,,
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (74.0.0)
2
+ Generator: setuptools (75.1.0)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5
 
@@ -1,13 +0,0 @@
1
- thop/__init__.py,sha256=sjz2M_shMkJ2HPv4v-qPjrGbMmsaQilGjBCc0RbcgGY,146
2
- thop/fx_profile.py,sha256=u1eNvsRBTl9u0jJ6ymgrnrJP1VSLpAZkaEvm4D2mrI4,8172
3
- thop/profile.py,sha256=7ZMuGia9jIz_aeuUWXD_wb5WgunuaPx7nfaUQVf8CzY,7911
4
- thop/rnn_hooks.py,sha256=IwxLvVvbeo5skThdg3BKXm08_0GsM8FI2QM_alQcCB8,6432
5
- thop/utils.py,sha256=SP5QWanMqVRieDoYGQN-7lftZmYXohn2E_m3cP17tS4,1413
6
- thop/vision/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
7
- thop/vision/basic_hooks.py,sha256=tKclhcddDByXrIXiI9u224nIKTbqWfp2J3hqIUYK6Hc,4639
8
- thop/vision/calc_func.py,sha256=5YbWp48XRbJYzSRkAFMr5xlM4o5BbjTcU6rHSKPkD6I,4291
9
- ultralytics_thop-2.0.6.dist-info/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
10
- ultralytics_thop-2.0.6.dist-info/METADATA,sha256=IISDJzN75x5aachcT7wtKf6dclFMsR3HoNmJFjXUiGM,9123
11
- ultralytics_thop-2.0.6.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
12
- ultralytics_thop-2.0.6.dist-info/top_level.txt,sha256=HQ7D0gSvDJ31CNR-f0EuXNVve05RYBmwyIkHQKiEhU8,5
13
- ultralytics_thop-2.0.6.dist-info/RECORD,,