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