Update centernet2-cascade_r50_fpn_1x_coco.py

open-mmlab · Jan 6, 2023 · 497505c · 497505c
1 parent 6ad0b8e
commit 497505c
Showing 1 changed file with 2 additions and 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
@@ -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