ultralytics-thop 2.0.6__tar.gz → 2.0.8__tar.gz
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.
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/PKG-INFO +7 -5
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/README.md +1 -1
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/pyproject.toml +5 -3
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/thop/__init__.py +2 -1
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/thop/profile.py +2 -2
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/thop/vision/basic_hooks.py +1 -1
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/ultralytics_thop.egg-info/PKG-INFO +7 -5
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/LICENSE +0 -0
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/setup.cfg +0 -0
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/tests/test_conv2d.py +0 -0
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/tests/test_matmul.py +0 -0
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/tests/test_relu.py +0 -0
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/tests/test_utils.py +0 -0
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/thop/fx_profile.py +0 -0
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/thop/rnn_hooks.py +0 -0
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/thop/utils.py +0 -0
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/thop/vision/__init__.py +0 -0
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/thop/vision/calc_func.py +0 -0
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/ultralytics_thop.egg-info/SOURCES.txt +0 -0
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/ultralytics_thop.egg-info/dependency_links.txt +0 -0
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/ultralytics_thop.egg-info/requires.txt +0 -0
- {ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/ultralytics_thop.egg-info/top_level.txt +0 -0
|
@@ -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">
|
|
@@ -141,7 +141,7 @@ THOP is licensed under the AGPL-3.0 License. For more information, see the [LICE
|
|
|
141
141
|
|
|
142
142
|
## 📮 Contact
|
|
143
143
|
|
|
144
|
-
For bugs or feature requests, please open an issue on [GitHub Issues](https://github.com/ultralytics/thop/
|
|
144
|
+
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.
|
|
145
145
|
|
|
146
146
|
<br>
|
|
147
147
|
<div align="center">
|
|
@@ -35,7 +35,7 @@ authors = [
|
|
|
35
35
|
{ name = "Ligeng Zhu", email = "ligeng.zhu+github@gmail.com" }
|
|
36
36
|
]
|
|
37
37
|
maintainers = [
|
|
38
|
-
{ name = "
|
|
38
|
+
{ name = "Ultralytics", email = "hello@ultralytics.com" }
|
|
39
39
|
]
|
|
40
40
|
classifiers = [
|
|
41
41
|
"Development Status :: 4 - Beta",
|
|
@@ -63,9 +63,11 @@ dependencies = [
|
|
|
63
63
|
]
|
|
64
64
|
|
|
65
65
|
[project.urls]
|
|
66
|
+
"Homepage" = "https://ultralytics.com"
|
|
67
|
+
"Source" = "https://github.com/ultralytics/thop"
|
|
68
|
+
"Documentation" = "https://docs.ultralytics.com"
|
|
66
69
|
"Bug Reports" = "https://github.com/ultralytics/thop/issues"
|
|
67
|
-
"
|
|
68
|
-
"Source" = "https://github.com/ultralytics/thop/"
|
|
70
|
+
"Changelog" = "https://github.com/ultralytics/thop/releases"
|
|
69
71
|
|
|
70
72
|
[tool.setuptools.dynamic]
|
|
71
73
|
version = { attr = "thop.__version__" }
|
|
@@ -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))
|
|
@@ -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">
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
{ultralytics_thop-2.0.6 → ultralytics_thop-2.0.8}/ultralytics_thop.egg-info/dependency_links.txt
RENAMED
|
File without changes
|
|
File without changes
|
|
File without changes
|