From 857bac74cfa85fe2ca230867c2b368feedec7114 Mon Sep 17 00:00:00 2001
From: fhw <fhw@qq.com>
Date: Fri, 12 Jul 2024 10:51:35 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E4=BF=9D=E5=AD=98=E6=9D=83?=
 =?UTF-8?q?=E9=87=8D=E9=97=B4=E9=9A=94=E5=92=8C=E8=AE=AD=E7=BB=83=E8=BD=AE?=
 =?UTF-8?q?=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../topdown_heatmap/coco/ViTSam_base_coco_256x192.py          | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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,