Home

Awesome

Accelerated Multi-Modal MR Imaging with Transformers

Dependencies

:fire: NEWS :fire:

multi gpu train

python -m torch.distributed.launch --nproc_per_node=8   train.py --experiment sr_multi_cross

single gpu train

python train.py --experiment sr_multi_cross

multi gpu test

python -m torch.distributed.launch --nproc_per_node=8   test.py --experiment sr_multi_cross

single gpu test

python test.py --experiment sr_multi_cross

--experiment is the experiment you running. And you can change the config file to set the parameter when training or testing.

Citation

@article{feng2021accelerated,
  title={Accelerated Multi-Modal MR Imaging with Transformers},
  author={Feng, Chun-Mei and Yan, Yunlu and Chen, Geng and Fu, Huazhu and Xu, Yong and Shao, Ling},
  journal={arXiv e-prints},
  pages={arXiv--2106},
  year={2021}
}