From 70de0899be7a9e99be70f3b2716b805dada1bdb1 Mon Sep 17 00:00:00 2001
From: zytx121 <592267829@qq.com>
Date: Fri, 6 Jan 2023 22:35:01 +0800
Subject: [PATCH] Update centernet2-cascade_r50_fpn_1x_coco.py

---
 .../configs/centernet2-cascade_r50_fpn_1x_coco.py           | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/projects/CenterNet2/configs/centernet2-cascade_r50_fpn_1x_coco.py b/projects/CenterNet2/configs/centernet2-cascade_r50_fpn_1x_coco.py
index cb06ef71fa7..968ff974c3b 100644
--- a/projects/CenterNet2/configs/centernet2-cascade_r50_fpn_1x_coco.py
+++ b/projects/CenterNet2/configs/centernet2-cascade_r50_fpn_1x_coco.py
@@ -199,8 +199,7 @@
 ]
 
 # optimizer
-optim_wrapper = dict(
-    optimizer=dict(lr=0.04), clip_grad=dict(max_norm=35, norm_type=2))
+optim_wrapper = dict(clip_grad=dict(max_norm=35, norm_type=2))
 
 train_pipeline = [
     dict(
@@ -216,5 +215,4 @@
     dict(type='PackDetInputs')
 ]
 
-train_dataloader = dict(
-    batch_size=8, num_workers=4, dataset=dict(pipeline=train_pipeline))
+find_unused_parameters = True