update

Tau-J · Oct 30, 2023 · db020d2 · db020d2
1 parent 6813ca1
commit db020d2
Show file tree

Hide file tree

Showing 7 changed files with 2,934 additions and 3 deletions.
diff --git a/configs/wholebody_2d_keypoint/rtmpose/wholebody8/wb_d1017rc1.py b/configs/wholebody_2d_keypoint/rtmpose/wholebody8/wb_d1017rc1.py
@@ -48,7 +48,7 @@
 # optimizer
 optim_wrapper = dict(
     type=OptimWrapper,
-    optimizer=dict(type=AdamW, lr=base_lr, weight_decay=0.05),
+    optimizer=dict(type=AdamW, lr=base_lr, weight_decay=0.1),
     clip_grad=dict(max_norm=35, norm_type=2),
     paramwise_cfg=dict(
         norm_decay_mult=0, bias_decay_mult=0, bypass_duplicate=True))

diff --git a/configs/wholebody_2d_keypoint/rtmpose/wholebody8/wb_d1023rc1.py b/configs/wholebody_2d_keypoint/rtmpose/wholebody8/wb_d1023rc1.py
@@ -48,7 +48,7 @@
 # optimizer
 optim_wrapper = dict(
     type=OptimWrapper,
-    optimizer=dict(type=AdamW, lr=base_lr, weight_decay=0.05),
+    optimizer=dict(type=AdamW, lr=base_lr, weight_decay=0.1),
     clip_grad=dict(max_norm=35, norm_type=2),
     paramwise_cfg=dict(
         norm_decay_mult=0, bias_decay_mult=0, bypass_duplicate=True))
@@ -134,7 +134,9 @@
         loss=dict(
             type=KLDiscretLoss,
             use_target_weight=True,
-            beta=10.,
+            beta=1.,
+            mask=list(range(23, 91)),
+            mask_weight=0.5,
             label_softmax=True),
         decoder=codec),
     test_cfg=dict(flip_test=True))