更新保存权重间隔和训练轮数

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,