| cd ../; | |
| CUDA_VISIBLE_DEVICES=0 torchrun main_ecg_2021.py \ | |
| --model ECG_VMamba_architecture \ | |
| --batch-size 20 \ | |
| --num_workers 4 \ | |
| --mixup 0 \ | |
| --cutmix 0 \ | |
| --mixup_no_label 0 \ | |
| --epochs 60 \ | |
| --lead 12Lead \ | |
| --block VMamba \ | |
| --depth 14 \ | |
| --lrschedule CosineAnnealing | |