align weight decay to the mmdet2

LutingWang · Feb 24, 2024 · 8dc3901 · 8dc3901
1 parent e93f8d9
commit 8dc3901
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/configs/dp/datasets/coco_detection.py b/configs/dp/datasets/coco_detection.py
@@ -37,7 +37,7 @@
     )
 ]
 train_dataloader = dict(
-    batch_size=4,
+    batch_size=2,
     num_workers=2,
     persistent_workers=True,
     sampler=dict(type='DefaultSampler', shuffle=True),

diff --git a/configs/dp/schedules/2x.py b/configs/dp/schedules/2x.py
@@ -21,7 +21,7 @@
 # optimizer
 optim_wrapper = dict(
     type='OptimWrapper',
-    optimizer=dict(type='SGD', lr=0.02, momentum=0.9, weight_decay=0.0001)
+    optimizer=dict(type='SGD', lr=0.02, momentum=0.9, weight_decay=2.5e-5)
 )
 
 # Default setting for scaling LR automatically

diff --git a/configs/dp/schedules/40k.py b/configs/dp/schedules/40k.py
@@ -25,7 +25,7 @@
 # optimizer
 optim_wrapper = dict(
     type='OptimWrapper',
-    optimizer=dict(type='SGD', lr=0.02, momentum=0.9, weight_decay=0.0001)
+    optimizer=dict(type='SGD', lr=0.02, momentum=0.9, weight_decay=2.5e-5)
 )
 # Default setting for scaling LR automatically
 #   - `enable` means enable scaling LR automatically

diff --git a/oadp/dp/datasets.py b/oadp/dp/datasets.py
@@ -18,6 +18,7 @@
 
 BLOCK_BBOXES_FLAG = 2
 OBJECT_BBOXES_FLAG = 3
+np.float = np.float32  # type: ignore
 
 
 class DebugMixin(BaseDetDataset):