[cherry-pick]add clip grad in ppyolo config (#7353)

02ee2399 · wangxinxin08 · GitHub · da749333 · 02ee2399 · 02ee2399
隐藏空白更改
内联并排

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