MODEL_PATH = 'your_model_path' 意義?

[Dandelionlibra]

您好，
請問在 videomamba.py 下的這個路徑要如何修正呢，目前我正在嘗試運行 k400/videomamba_tiny/run_f8x224.sh，並有對 run_f8x224.sh 中的內容略為修改。

MODEL_PATH = 'your_model_path'
_MODELS = {
    "videomamba_t16_in1k": os.path.join(MODEL_PATH, "videomamba_t16_in1k_res224.pth"),
    "videomamba_s16_in1k": os.path.join(MODEL_PATH, "videomamba_s16_in1k_res224.pth"),
    "videomamba_m16_in1k": os.path.join(MODEL_PATH, "videomamba_m16_in1k_res224.pth"),
}

目前錯誤：FileNotFoundError: [Errno 2] No such file or directory: 'your_model_path/videomamba_t16_in1k_res224.pth'
是要使用以下路徑的預訓練參數嗎?
https://huggingface.co/OpenGVLab/VideoMamba/blob/main/videomamba_t16_k400_f8_res224.pth

此外，run_f8x224.sh 做如下修正是可行的嗎?

export MASTER_PORT=$((12000 + $RANDOM % 20000))
export OMP_NUM_THREADS=1
# 設定使用 GPU 0, 3090 ti
export CUDA_VISIBLE_DEVICES=0

JOB_NAME='videomamba_tiny_f8_res224_test'
OUTPUT_DIR="$(dirname "$0")/$JOB_NAME"
LOG_DIR="./logs/${JOB_NAME}"
# 設定數據(影片檔)的路徑
PREFIX='/data_in_nas/CASME_3/k400/kinetics-dataset/unzip_part_0'
# 設定數據集的路徑/train.csv
DATA_PATH='/data_in_docker/V/videomamba/video_sm/exp/k400'

# 設定本地環境
PARTITION='video5' # 設定合適的分區
GPUS=1 # 只有1顆GPU
GPUS_PER_NODE=1 # 每個節點的GPU數量
CPUS_PER_TASK=1 # 設定每個任務的CPU數量

# torchrun --nnodes=1 --nproc-per-node=1 \ 
python3 run_class_finetuning.py \
 --model videomamba_tiny \
 --data_path ${DATA_PATH} \
 --prefix ${PREFIX} \
 --data_set 'Kinetics_sparse' \
 --split ',' \
 --nb_classes 400 \
 --log_dir ${OUTPUT_DIR} \
 --output_dir ${OUTPUT_DIR} \
 --batch_size 8 \
 --num_sample 2 \
 --input_size 224 \
 --short_side_size 224 \
 --save_ckpt_freq 100 \
 --num_frames 8 \
 --num_workers 12 \
 --warmup_epochs 5 \
 --tubelet_size 1 \
 --epochs 45 \
 --lr 1e-4 \
 --layer_decay 0.8 \
 --drop_path 0.4 \
 --opt adamw \
 --opt_betas 0.9 0.999 \
 --weight_decay 0.05 \
 --test_num_segment 4 \
 --test_num_crop 3 \
 --dist_eval \
 --test_best \
 --bf16

麻煩您解惑了，謝謝。

[ThatIsMyUsername]

将'your_model_path'替换成你下载.pth后的路径