Refactor code for speed and clarity

README.md

@@ -155,7 +155,7 @@ For bugs or feature requests, please open an issue on [GitHub Issues](https://gi
   <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
   <a href="https://www.tiktok.com/@ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-tiktok.png" width="3%" alt="Ultralytics TikTok"></a>
   <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
-  <a href="https://ultralytics.com/bilibili"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-bilibili.png" width="3%" alt="Ultralytics Instagram"></a>
+  <a href="https://ultralytics.com/bilibili"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-bilibili.png" width="3%" alt="Ultralytics BiliBili"></a>
   <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
   <a href="https://ultralytics.com/discord"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-discord.png" width="3%" alt="Ultralytics Discord"></a>
 </div>

benchmark/evaluate_famous_models.py

@@ -1,7 +1,6 @@
 import torch
-from torchvision import models
-
 from thop.profile import profile
+from torchvision import models
 
 model_names = sorted(
     name

benchmark/evaluate_rnn_models.py

@@ -1,6 +1,5 @@
 import torch
 import torch.nn as nn
-
 from thop.profile import profile
 
 input_size = 160

tests/test_conv2d.py

@@ -1,6 +1,5 @@
 import torch
 import torch.nn as nn
-
 from thop import profile
 
 

tests/test_matmul.py

@@ -1,6 +1,5 @@
 import torch
 import torch.nn as nn
-
 from thop import profile
 
 

tests/test_relu.py

@@ -1,6 +1,5 @@
 import torch
 import torch.nn as nn
-
 from thop import profile