From 6d644a1d85793a2f74959977159f270eee95bf27 Mon Sep 17 00:00:00 2001
From: wangxinxin08 <69842442+wangxinxin08@users.noreply.github.com>
Date: Thu, 17 Nov 2022 17:56:33 +0800
Subject: [PATCH] add clip grad in ppyolo config (#7352)

---
 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