1 changed files with 12 additions and 0 deletions
@ -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…
Reference in new issue