junyuchen245 / TransMorph_Transformer_for_Medical_Image_Registration

TransMorph: Transformer for Unsupervised Medical Image Registration (PyTorch)
MIT License
437 stars 73 forks source link

parallel training #57

Open Clean1224 opened 1 year ago

Clean1224 commented 1 year ago

First of all, I would like to thank you for your code.I noticed that you have not used parallel training yet. I was wondering if you could provide a tutorial on how to use DistributedDataParallel for parallel training?