From 02ee239956c7e64c0cb02b4ceada44cb3b68f458 Mon Sep 17 00:00:00 2001
From: wangxinxin08 <69842442+wangxinxin08@users.noreply.github.com>
Date: Sat, 19 Nov 2022 16:34:13 +0800
Subject: [PATCH] [cherry-pick]add clip grad in ppyolo config (#7353)

---
 configs/ppyolo/_base_/optimizer_1x.yml | 1 +
 configs/ppyolo/_base_/optimizer_2x.yml | 1 +
 2 files changed, 2 insertions(+)

diff --git a/configs/ppyolo/_base_/optimizer_1x.yml b/configs/ppyolo/_base_/optimizer_1x.yml
index 8e6301e32..fe51b296c 100644
--- 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
diff --git a/configs/ppyolo/_base_/optimizer_2x.yml b/configs/ppyolo/_base_/optimizer_2x.yml
index 92ddbf2a7..c601a1860 100644
--- 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
-- 
GitLab