ultralytics 8.0.154 add freeze training argument (#4329)

2023-08-14 00:18:32 +02:00
parent 9f6d48d3cf
commit d47718c367
6 changed files with 120 additions and 98 deletions
--- a/ultralytics/nn/modules/conv.py
+++ b/ultralytics/nn/modules/conv.py
@ -142,8 +142,9 @@ class GhostConv(nn.Module):


 class RepConv(nn.Module):
-    """RepConv is a basic rep-style block, including training and deploy status
-    This code is based on https://github.com/DingXiaoH/RepVGG/blob/main/repvgg.py
+    """
+    RepConv is a basic rep-style block, including training and deploy status. This module is used in RT-DETR.
+    Based on https://github.com/DingXiaoH/RepVGG/blob/main/repvgg.py
    """
    default_act = nn.SiLU()  # default activation