jeya-maria-jose / UNeXt-pytorch

Official Pytorch Code base for "UNeXt: MLP-based Rapid Medical Image Segmentation Network", MICCAI 2022
https://jeya-maria-jose.github.io/UNext-web/
MIT License
459 stars 76 forks source link

select GPU? #29

Open Better-77 opened 1 year ago

Better-77 commented 1 year ago

How do I select a specific GPU to train my model?