From e058e90ef84426c4a5a483208ce749e1f1bb0902 Mon Sep 17 00:00:00 2001
From: fhw <fhw@qq.com>
Date: Sat, 13 Jul 2024 17:45:17 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E8=AE=AD=E7=BB=83shell?=
 =?UTF-8?q?=E8=84=9A=E6=9C=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 run_train.sh | 12 ++++++++++++
 1 file changed, 12 insertions(+)
 create mode 100644 run_train.sh

diff --git a/run_train.sh b/run_train.sh
new file mode 100644
index 0000000..7051516
--- /dev/null
+++ b/run_train.sh
@@ -0,0 +1,12 @@
+#!/bin/bash
+
+# Python 脚本路径
+PYTHON_SCRIPT="tools/train+sam.py"
+
+# 固定的参数
+CONFIG_FILE="configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py"
+WORK_DIR="work_dirs/ViTSam_base_coco_256x192_frozen_sam_ffn"
+RESUME="work_dirs/ViTSam_base_coco_256x192_frozen_sam_ffn/epoch_2.pth"
+
+# 执行 Python 脚本并传递固定的参数
+python $PYTHON_SCRIPT $CONFIG_FILE --work-dir $WORK_DIR --resume-from $RESUME
\ No newline at end of file