fix paramwise_cfg

open-mmlab · Feb 22, 2023 · 4fa4866 · 4fa4866
1 parent bef2736
commit 4fa4866
Show file tree

Hide file tree

Showing 6 changed files with 26 additions and 24 deletions.
diff --git a/configs/body_2d_keypoint/topdown_heatmap/coco/td-hm_hrformer-base_8xb32-210e_coco-256x192.py b/configs/body_2d_keypoint/topdown_heatmap/coco/td-hm_hrformer-base_8xb32-210e_coco-256x192.py
@@ -10,9 +10,9 @@
         lr=5e-4,
         betas=(0.9, 0.999),
         weight_decay=0.01,
-        paramwise_cfg=dict(
-            custom_keys={'relative_position_bias_table': dict(
-                decay_mult=0.)})))
+    ),
+    paramwise_cfg=dict(
+        custom_keys={'relative_position_bias_table': dict(decay_mult=0.)}))
 
 # learning policy
 param_scheduler = [

diff --git a/configs/body_2d_keypoint/topdown_heatmap/coco/td-hm_hrformer-base_8xb32-210e_coco-384x288.py b/configs/body_2d_keypoint/topdown_heatmap/coco/td-hm_hrformer-base_8xb32-210e_coco-384x288.py
@@ -10,9 +10,9 @@
         lr=5e-4,
         betas=(0.9, 0.999),
         weight_decay=0.01,
-        paramwise_cfg=dict(
-            custom_keys={'relative_position_bias_table': dict(
-                decay_mult=0.)})))
+    ),
+    paramwise_cfg=dict(
+        custom_keys={'relative_position_bias_table': dict(decay_mult=0.)}))
 
 # learning policy
 param_scheduler = [

diff --git a/...igs/body_2d_keypoint/topdown_heatmap/coco/td-hm_hrformer-small_8xb32-210e_coco-256x192.py b/...igs/body_2d_keypoint/topdown_heatmap/coco/td-hm_hrformer-small_8xb32-210e_coco-256x192.py
@@ -10,9 +10,9 @@
         lr=5e-4,
         betas=(0.9, 0.999),
         weight_decay=0.01,
-        paramwise_cfg=dict(
-            custom_keys={'relative_position_bias_table': dict(
-                decay_mult=0.)})))
+    ),
+    paramwise_cfg=dict(
+        custom_keys={'relative_position_bias_table': dict(decay_mult=0.)}))
 
 # learning policy
 param_scheduler = [

diff --git a/...igs/body_2d_keypoint/topdown_heatmap/coco/td-hm_hrformer-small_8xb32-210e_coco-384x288.py b/...igs/body_2d_keypoint/topdown_heatmap/coco/td-hm_hrformer-small_8xb32-210e_coco-384x288.py
@@ -10,9 +10,9 @@
         lr=5e-4,
         betas=(0.9, 0.999),
         weight_decay=0.01,
-        paramwise_cfg=dict(
-            custom_keys={'relative_position_bias_table': dict(
-                decay_mult=0.)})))
+    ),
+    paramwise_cfg=dict(
+        custom_keys={'relative_position_bias_table': dict(decay_mult=0.)}))
 
 # learning policy
 param_scheduler = [

diff --git a/configs/body_2d_keypoint/topdown_heatmap/coco/td-hm_swin-l-p4-w7_8xb32-210e_coco-256x192.py b/configs/body_2d_keypoint/topdown_heatmap/coco/td-hm_swin-l-p4-w7_8xb32-210e_coco-256x192.py
@@ -10,12 +10,13 @@
         lr=5e-4,
         betas=(0.9, 0.999),
         weight_decay=0.01,
-        paramwise_cfg=dict(
-            custom_keys={
-                'absolute_pos_embed': dict(decay_mult=0.),
-                'relative_position_bias_table': dict(decay_mult=0.),
-                'norm': dict(decay_mult=0.)
-            })))
+    ),
+    paramwise_cfg=dict(
+        custom_keys={
+            'absolute_pos_embed': dict(decay_mult=0.),
+            'relative_position_bias_table': dict(decay_mult=0.),
+            'norm': dict(decay_mult=0.)
+        }))
 
 # learning policy
 param_scheduler = [

diff --git a/configs/body_2d_keypoint/topdown_heatmap/coco/td-hm_swin-l-p4-w7_8xb32-210e_coco-384x288.py b/configs/body_2d_keypoint/topdown_heatmap/coco/td-hm_swin-l-p4-w7_8xb32-210e_coco-384x288.py
@@ -10,12 +10,13 @@
         lr=5e-4,
         betas=(0.9, 0.999),
         weight_decay=0.01,
-        paramwise_cfg=dict(
-            custom_keys={
-                'absolute_pos_embed': dict(decay_mult=0.),
-                'relative_position_bias_table': dict(decay_mult=0.),
-                'norm': dict(decay_mult=0.)
-            })))
+    ),
+    paramwise_cfg=dict(
+        custom_keys={
+            'absolute_pos_embed': dict(decay_mult=0.),
+            'relative_position_bias_table': dict(decay_mult=0.),
+            'norm': dict(decay_mult=0.)
+        }))
 
 # learning policy
 param_scheduler = [