diff --git a/configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py b/configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py index 3f7cf86..60a1108 100644 --- a/configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py +++ b/configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py @@ -4,6 +4,8 @@ _base_ = [ ] evaluation = dict(interval=1, metric='mAP', save_best='AP') +checkpoint_config = dict(interval=1) # 保存模型权重文件的间隔 + optimizer = dict(type='AdamW', lr=5e-4, betas=(0.9, 0.999), weight_decay=0.1, constructor='LayerDecayOptimizerConstructor', @@ -28,7 +30,7 @@ lr_config = dict( warmup_iters=500, warmup_ratio=0.001, step=[170, 200]) -total_epochs = 210 +total_epochs = 20 target_type = 'GaussianHeatmap' channel_cfg = dict( num_output_channels=17,