add clip grad

yolo/config/gelan_config.py

@@ -90,7 +90,7 @@ class GElanBaseConfig(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   = -1.
+        self.clip_max_norm   = 35.0
         self.warmup_bias_lr  = 0.1
         self.warmup_momentum = 0.8
 

yolo/config/yolov1_config.py

@@ -58,7 +58,7 @@ class Yolov1BaseConfig(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   = -1.
+        self.clip_max_norm   = 35.0
         self.warmup_bias_lr  = 0.1
         self.warmup_momentum = 0.8
 

yolo/config/yolov2_config.py

@@ -61,7 +61,7 @@ class Yolov2BaseConfig(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   = -1.
+        self.clip_max_norm   = 35.0
         self.warmup_bias_lr  = 0.1
         self.warmup_momentum = 0.8
 

yolo/config/yolov5_af_config.py

@@ -78,7 +78,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   = -1.
+        self.clip_max_norm   = 35.0
         self.warmup_bias_lr  = 0.1
         self.warmup_momentum = 0.8
 

yolo/config/yolov5_config.py

@@ -80,7 +80,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   = -1.
+        self.clip_max_norm   = 35.0
         self.warmup_bias_lr  = 0.1
         self.warmup_momentum = 0.8
 

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   = -1.
+        self.clip_max_norm   = 35.0
         self.warmup_bias_lr  = 0.1
         self.warmup_momentum = 0.8
 

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   = -1.
+        self.clip_max_norm   = 35.0
         self.warmup_bias_lr  = 0.1
         self.warmup_momentum = 0.8