cd ../;

CUDA_VISIBLE_DEVICES=0 torchrun main_ecg.py \
--model ecg_vim_small_patch16_stride8_224_bimambav2_final_pool_mean_abs_pos_embed_with_midclstok_div2 \
--batch-size 20 \
--num_workers 4 \
--mixup 0 \
--cutmix 0 \
--mixup_no_label 0 \
--epochs 60 \
--lead 12Lead \
--block VisionMamba \
--depth 24 \
--lrschedule Noam