Refactor code for speed and clarity

ultralytics · Jan 15, 2025 · b5d56c3 · b5d56c3
1 parent 630da13
commit b5d56c3
Show file tree

Hide file tree

Showing 8 changed files with 7 additions and 9 deletions.
diff --git a/.github/workflows/format.yml b/.github/workflows/format.yml
@@ -1,4 +1,5 @@
-# Ultralytics 🚀 - AGPL-3.0 License https://ultralytics.com/license
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
 # Ultralytics Actions https://github.com/ultralytics/actions
 # This workflow automatically formats code and documentation in PRs to official Ultralytics standards
 

diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -1,4 +1,5 @@
-# Ultralytics YOLO 🚀, AGPL-3.0 license
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
 # Publish pip package to PyPI https://pypi.org/project/ultralytics-thop/
 
 name: Publish to PyPI

diff --git a/benchmark/evaluate_famous_models.py b/benchmark/evaluate_famous_models.py
@@ -1,9 +1,8 @@
 # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
 
 import torch
-from torchvision import models
-
 from thop.profile import profile
+from torchvision import models
 
 model_names = sorted(
     name

diff --git a/benchmark/evaluate_rnn_models.py b/benchmark/evaluate_rnn_models.py
@@ -2,7 +2,6 @@
 
 import torch
 import torch.nn as nn
-
 from thop.profile import profile
 
 input_size = 160

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,4 +1,5 @@
-# Ultralytics YOLO 🚀, AGPL-3.0 license
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
 # Publish pip package to PyPI https://pypi.org/project/ultralytics/ and Docs to https://docs.ultralytics.com
 #
 # Overview:

diff --git a/tests/test_conv2d.py b/tests/test_conv2d.py
@@ -2,7 +2,6 @@
 
 import torch
 import torch.nn as nn
-
 from thop import profile
 
 

diff --git a/tests/test_matmul.py b/tests/test_matmul.py
@@ -2,7 +2,6 @@
 
 import torch
 import torch.nn as nn
-
 from thop import profile
 
 

diff --git a/tests/test_relu.py b/tests/test_relu.py
@@ -2,7 +2,6 @@
 
 import torch
 import torch.nn as nn
-
 from thop import profile
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2,7 +2,6 @@

		import torch
		import torch.nn as nn

		from thop import profile


Expand Down