上传训练脚本

1 year ago · 27cebb946f
1 changed files with 2 additions and 2 deletions
--- a/run_train.sh
+++ b/run_train.sh
@ -5,8 +5,8 @@ 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"
+WORK_DIR="work_dirs/ViTSam_base_coco_256x192_frozen_cross-attn"
 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
+python $PYTHON_SCRIPT $CONFIG_FILE --work-dir $WORK_DIR