Browse Source

添加训练shell脚本

main
fhw 1 year ago
parent
commit
e058e90ef8
  1. 12
      run_train.sh

12
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
Loading…
Cancel
Save