| 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 | |