ultralytics-thop 0.2.7__py3-none-any.whl → 0.2.8__py3-none-any.whl

thop/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.2.7"
+__version__ = "0.2.8"
 
 import torch
 

thop/fx_profile.py

@@ -148,7 +148,6 @@ def fx_profile(mod: nn.Module, input: th.Tensor, verbose=False):
         node_flops = None
 
         input_shapes = []
-        output_shapes = []
         fprint("input_shape:", end="\t")
         for arg in node.args:
             if str(arg) not in v_maps:
@@ -157,8 +156,7 @@ def fx_profile(mod: nn.Module, input: th.Tensor, verbose=False):
             input_shapes.append(v_maps[str(arg)])
         fprint()
         fprint(f"output_shape:\t{node.meta['tensor_meta'].shape}")
-        output_shapes.append(node.meta["tensor_meta"].shape)
-
+        output_shapes = [node.meta["tensor_meta"].shape]
         if node.op in ["output", "placeholder"]:
             node_flops = 0
         elif node.op == "call_function":
@@ -194,7 +192,7 @@ def fx_profile(mod: nn.Module, input: th.Tensor, verbose=False):
                 print("weight_shape: None")
             else:
                 print(type(m))
-                print(f"weight_shape: {mod.state_dict()[node.target + '.weight'].shape}")
+                print(f"weight_shape: {mod.state_dict()[f'{node.target}.weight'].shape}")
 
         v_maps[str(node.name)] = node.meta["tensor_meta"].shape
         if node_flops is not None:

thop/vision/basic_hooks.py

@@ -10,9 +10,7 @@ multiply_adds = 1
 
 def count_parameters(m, x, y):
     """Calculate and update the total number of parameters in a given PyTorch model."""
-    total_params = 0
-    for p in m.parameters():
-        total_params += torch.DoubleTensor([p.numel()])
+    total_params = sum(torch.DoubleTensor([p.numel()]) for p in m.parameters())
     m.total_params[0] = calculate_parameters(m.parameters())
 
 

{ultralytics_thop-0.2.7.dist-info → ultralytics_thop-0.2.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics-thop
-Version: 0.2.7
+Version: 0.2.8
 Summary: Ultralytics THOP package for fast computation of PyTorch model FLOPs and parameters.
 Author-email: Ligeng Zhu <ligeng.zhu+github@gmail.com>
 Maintainer: Glenn Jocher
@@ -74,7 +74,7 @@ import torch
 
 model = resnet50()
 input = torch.randn(1, 3, 224, 224)
-macs, params = profile(model, inputs=(input, ))
+macs, params = profile(model, inputs=(input,))
 ```
 
 ### Define Custom Rules for Third-Party Modules
@@ -84,16 +84,19 @@ You can define custom rules for unsupported modules:
 ```python
 import torch.nn as nn
 
+
 class YourModule(nn.Module):
     # your definition
     pass
 
+
 def count_your_model(model, x, y):
     # your rule here
     pass
 
+
 input = torch.randn(1, 3, 224, 224)
-macs, params = profile(model, inputs=(input, ), custom_ops={YourModule: count_your_model})
+macs, params = profile(model, inputs=(input,), custom_ops={YourModule: count_your_model})
 ```
 
 ### Improve Output Readability
@@ -102,6 +105,7 @@ Use `thop.clever_format` for a more readable output:
 
 ```python
 from thop import clever_format
+
 macs, params = clever_format([macs, params], "%.3f")
 ```
 

ultralytics_thop-0.2.8.dist-info/RECORD

@@ -0,0 +1,13 @@
+thop/__init__.py,sha256=nRitFGQJlu5vCs1JrMHYbmf8_rIcIfJ6ujcV9--_dOM,146
+thop/fx_profile.py,sha256=DSC3phrsDj08tey_OYVeiL6A8q3BE1er2YjH8W60VrM,8238
+thop/profile.py,sha256=z89mX1zVr_42axKbhgz9k2MJbgBLaBPt30lYT_PcWuA,7848
+thop/rnn_hooks.py,sha256=GYuKaNPEdZzTuqNCrJpTBjeQHsWs65UXCCaViS2giik,6485
+thop/utils.py,sha256=V_Pj_qC6RjqMyuiSX05eeUujT07hZQV-xobCIabjDds,1422
+thop/vision/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+thop/vision/basic_hooks.py,sha256=i_CSCC0o5olEbK2hOirehDKfD8ae19IjCkyHk0Zr4ko,4839
+thop/vision/calc_func.py,sha256=RE-qQWGjZIlRx8CNKjF-ZY7aS0WShPkvnOAYvQA8z8I,4130
+ultralytics_thop-0.2.8.dist-info/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+ultralytics_thop-0.2.8.dist-info/METADATA,sha256=rIGQaLv_zrB6KB-qd35Z9L8igXLHf6xab4McW8d-Yhk,8525
+ultralytics_thop-0.2.8.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+ultralytics_thop-0.2.8.dist-info/top_level.txt,sha256=HQ7D0gSvDJ31CNR-f0EuXNVve05RYBmwyIkHQKiEhU8,5
+ultralytics_thop-0.2.8.dist-info/RECORD,,

ultralytics_thop-0.2.7.dist-info/RECORD

@@ -1,13 +0,0 @@
-thop/__init__.py,sha256=FI_uI0-brlbiGcvcf7nDnqAa1utJ2LN4Q2DE484cDcI,146
-thop/fx_profile.py,sha256=0xMPn0jsH4cpXT-8bd4cJP5gc2P6DudsuwmCoJ2Y48o,8270
-thop/profile.py,sha256=z89mX1zVr_42axKbhgz9k2MJbgBLaBPt30lYT_PcWuA,7848
-thop/rnn_hooks.py,sha256=GYuKaNPEdZzTuqNCrJpTBjeQHsWs65UXCCaViS2giik,6485
-thop/utils.py,sha256=V_Pj_qC6RjqMyuiSX05eeUujT07hZQV-xobCIabjDds,1422
-thop/vision/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-thop/vision/basic_hooks.py,sha256=RwJR5dNjDMEF9HaX077E6eEH1P90JIGPzel8H2A-y2k,4865
-thop/vision/calc_func.py,sha256=RE-qQWGjZIlRx8CNKjF-ZY7aS0WShPkvnOAYvQA8z8I,4130
-ultralytics_thop-0.2.7.dist-info/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-ultralytics_thop-0.2.7.dist-info/METADATA,sha256=7U3cuTQeP8IXMABNCXdmXzvOOa82ina-HPbgeNkEDwI,8523
-ultralytics_thop-0.2.7.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-ultralytics_thop-0.2.7.dist-info/top_level.txt,sha256=HQ7D0gSvDJ31CNR-f0EuXNVve05RYBmwyIkHQKiEhU8,5
-ultralytics_thop-0.2.7.dist-info/RECORD,,