From ae03a3238ac20d6bd3c63f47dfad6f0ec61e9f4e Mon Sep 17 00:00:00 2001
From: fhw <fhw@qq.com>
Date: Tue, 6 Aug 2024 19:44:01 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E8=AE=AD=E7=BB=83=E8=84=9A?=
 =?UTF-8?q?=E6=9C=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 run_train.sh | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/run_train.sh b/run_train.sh
index 091f12d..64b40e4 100644
--- a/run_train.sh
+++ b/run_train.sh
@@ -1,12 +1,15 @@
 #!/bin/bash
 
 # Python 脚本路径
-PYTHON_SCRIPT="tools/train+sam.py"
+PYTHON_SCRIPT="tools/train.py"
 
 # 固定的参数
-CONFIG_FILE="configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py"
-WORK_DIR="work_dirs/ViTPose_SAM-B_6"
-# RESUME="work_dirs/ViTSam_base_coco_256x192_frozen_sam_ffn/epoch_2.pth"
+CONFIG_FILE="configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTPose_base_coco_256x192.py"
+WORK_DIR="work_dirs/ViTPose-B_1"
+RESUME="work_dirs/ViTPose-B_1/epoch_30.pth"
+CFG_OPTIONS="model.pretrained=checkpoints/vit-pretrained/mae_pretrain_vit_base.pth"
 
 # 执行 Python 脚本并传递固定的参数
-python $PYTHON_SCRIPT $CONFIG_FILE --work-dir $WORK_DIR
\ No newline at end of file
+python $PYTHON_SCRIPT $CONFIG_FILE --work-dir $WORK_DIR \
+    --cfg-options $CFG_OPTIONS \
+    --resume-from $RESUME
\ No newline at end of file