yjh0410 1 год назад
Родитель
Сommit
318ddceab3

+ 1 - 1
yolo/config/yolov5_af_config.py

@@ -79,7 +79,7 @@ class Yolov5AFBaseConfig(object):
         self.min_lr_ratio = 0.01      # min_lr  = base_lr * min_lr_ratio
         self.momentum     = 0.9
         self.weight_decay = 0.05
-        self.clip_max_norm   = 35.0
+        self.clip_max_norm   = 10.0
         self.warmup_bias_lr  = 0.1
         self.warmup_momentum = 0.8
 

+ 1 - 1
yolo/config/yolov5_config.py

@@ -81,7 +81,7 @@ class Yolov5BaseConfig(object):
         self.min_lr_ratio = 0.01      # min_lr  = base_lr * min_lr_ratio
         self.momentum     = 0.9
         self.weight_decay = 0.05
-        self.clip_max_norm   = 35.0
+        self.clip_max_norm   = 10.0
         self.warmup_bias_lr  = 0.1
         self.warmup_momentum = 0.8
 

+ 1 - 1
yolo/config/yolov7_af_config.py

@@ -75,7 +75,7 @@ class Yolov7AFBaseConfig(object):
         self.min_lr_ratio = 0.01      # min_lr  = base_lr * min_lr_ratio
         self.momentum     = 0.9
         self.weight_decay = 0.05
-        self.clip_max_norm   = 35.0
+        self.clip_max_norm   = 10.0
         self.warmup_bias_lr  = 0.1
         self.warmup_momentum = 0.8
 

+ 1 - 1
yolo/config/yolov8_config.py

@@ -81,7 +81,7 @@ class Yolov8BaseConfig(object):
         self.min_lr_ratio = 0.01      # min_lr  = base_lr * min_lr_ratio
         self.momentum     = 0.9
         self.weight_decay = 0.05
-        self.clip_max_norm   = 35.0
+        self.clip_max_norm   = 10.0
         self.warmup_bias_lr  = 0.1
         self.warmup_momentum = 0.8