This repository is based on PyTorch 1.8.0, CUDA 11.3 and Python 3.8.10. All experiments in our paper were conducted on NVIDIA Quadro RTX 6000 24GB GPU with an identical experimental setting.
To train a model,
python ./code/ACDC_M3HL_train.py #for ACDC training
python ./code/LA_M3HL_train.py #for LA training
To test a model,
python ./code/test_ACDC.py #for ACDC testing
python ./code/test_LA.py #for LA testing