add clip grad in ppyolo config (#7352)

6d644a1d · wangxinxin08 · GitHub · 1d6b0a3d · 6d644a1d · 6d644a1d
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

configs/ppyolo/_base_/optimizer_1x.yml configs/ppyolo/_base_/optimizer_1x.yml +1 -0

configs/ppyolo/_base_/optimizer_2x.yml configs/ppyolo/_base_/optimizer_2x.yml +1 -0

未找到文件。
--- a/configs/ppyolo/_base_/optimizer_1x.yml
+++ b/configs/ppyolo/_base_/optimizer_1x.yml
@@ -13,6 +13,7 @@ LearningRate:
    steps: 4000

 OptimizerBuilder:
+  clip_grad_by_norm: 35.
  optimizer:
    momentum: 0.9
    type: Momentum

--- a/configs/ppyolo/_base_/optimizer_2x.yml
+++ b/configs/ppyolo/_base_/optimizer_2x.yml
@@ -13,6 +13,7 @@ LearningRate:
    steps: 4000

 OptimizerBuilder:
+  clip_grad_by_norm: 35.
  optimizer:
    momentum: 0.9
    type: Momentum