#!/bin/bash

# Python 脚本路径
PYTHON_SCRIPT="tools/train.py"

# 固定的参数
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 \
    --cfg-options $CFG_OPTIONS \
    --resume-from $RESUME