From 4fd8507ad3b1d1c7e0cd1491cf3f4ca82c35204f Mon Sep 17 00:00:00 2001 From: Yufei Date: Mon, 16 Jan 2023 20:00:04 +0800 Subject: [PATCH] update training params for vitpose-s --- ...e_huge_ap36k_256x192.py => ViTPose_huge_apt36k_256x192.py} | 0 .../topdown_heatmap/coco/ViTPose_small_coco_256x192.py | 4 ++-- .../topdown_heatmap/coco/ViTPose_small_simple_coco_256x192.py | 4 ++-- ..._small_coco+aic+mpii+ap10k+apt36k+wholebody_256x192_udp.py | 4 ++-- 4 files changed, 6 insertions(+), 6 deletions(-) rename configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/apt36k/{ViTPose_huge_ap36k_256x192.py => ViTPose_huge_apt36k_256x192.py} (100%) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/apt36k/ViTPose_huge_ap36k_256x192.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/apt36k/ViTPose_huge_apt36k_256x192.py similarity index 100% rename from configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/apt36k/ViTPose_huge_ap36k_256x192.py rename to configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/apt36k/ViTPose_huge_apt36k_256x192.py diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTPose_small_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTPose_small_coco_256x192.py index bec19fa..42ac25c 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTPose_small_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTPose_small_coco_256x192.py @@ -8,7 +8,7 @@ optimizer = dict(type='AdamW', lr=5e-4, betas=(0.9, 0.999), weight_decay=0.1, constructor='LayerDecayOptimizerConstructor', paramwise_cfg=dict( num_layers=12, - layer_decay_rate=0.9, + layer_decay_rate=0.8, custom_keys={ 'bias': dict(decay_multi=0.), 'pos_embed': dict(decay_mult=0.), @@ -54,7 +54,7 @@ model = dict( use_checkpoint=False, mlp_ratio=4, qkv_bias=True, - drop_path_rate=0.3, + drop_path_rate=0.1, ), keypoint_head=dict( type='TopdownHeatmapSimpleHead', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTPose_small_simple_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTPose_small_simple_coco_256x192.py index bec19fa..42ac25c 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTPose_small_simple_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTPose_small_simple_coco_256x192.py @@ -8,7 +8,7 @@ optimizer = dict(type='AdamW', lr=5e-4, betas=(0.9, 0.999), weight_decay=0.1, constructor='LayerDecayOptimizerConstructor', paramwise_cfg=dict( num_layers=12, - layer_decay_rate=0.9, + layer_decay_rate=0.8, custom_keys={ 'bias': dict(decay_multi=0.), 'pos_embed': dict(decay_mult=0.), @@ -54,7 +54,7 @@ model = dict( use_checkpoint=False, mlp_ratio=4, qkv_bias=True, - drop_path_rate=0.3, + drop_path_rate=0.1, ), keypoint_head=dict( type='TopdownHeatmapSimpleHead', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/vitPose+_small_coco+aic+mpii+ap10k+apt36k+wholebody_256x192_udp.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/vitPose+_small_coco+aic+mpii+ap10k+apt36k+wholebody_256x192_udp.py index 34823a3..0617aaa 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/vitPose+_small_coco+aic+mpii+ap10k+apt36k+wholebody_256x192_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/vitPose+_small_coco+aic+mpii+ap10k+apt36k+wholebody_256x192_udp.py @@ -12,7 +12,7 @@ optimizer = dict(type='AdamW', lr=1e-3, betas=(0.9, 0.999), weight_decay=0.1, constructor='LayerDecayOptimizerConstructor', paramwise_cfg=dict( num_layers=12, - layer_decay_rate=0.9, + layer_decay_rate=0.8, custom_keys={ 'bias': dict(decay_multi=0.), 'pos_embed': dict(decay_mult=0.), @@ -94,7 +94,7 @@ model = dict( use_checkpoint=False, mlp_ratio=4, qkv_bias=True, - drop_path_rate=0.3, + drop_path_rate=0.1, num_expert=6, part_features=192 ),