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 +2 -1
- thop/profile.py +2 -2
- thop/vision/basic_hooks.py +1 -1
- {ultralytics_thop-2.0.6.dist-info → ultralytics_thop-2.0.8.dist-info}/METADATA +7 -5
- ultralytics_thop-2.0.8.dist-info/RECORD +13 -0
- {ultralytics_thop-2.0.6.dist-info → ultralytics_thop-2.0.8.dist-info}/WHEEL +1 -1
- ultralytics_thop-2.0.6.dist-info/RECORD +0 -13
- {ultralytics_thop-2.0.6.dist-info → ultralytics_thop-2.0.8.dist-info}/LICENSE +0 -0
- {ultralytics_thop-2.0.6.dist-info → ultralytics_thop-2.0.8.dist-info}/top_level.txt +0 -0
thop/__init__.py
CHANGED
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
|
|
73
|
-
"Be careful, it might change your code's behavior."
|
|
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))
|
thop/vision/basic_hooks.py
CHANGED
|
@@ -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(
|
|
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.
|
|
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:
|
|
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:
|
|
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/
|
|
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,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,,
|
|
File without changes
|
|
File without changes
|