fix a bug in CPU mode - Githubissues

mattmacy / vnet.pytorch

A PyTorch implementation for V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation

https://mattmacy.github.io/vnet.pytorch

BSD 3-Clause "New" or "Revised" License

693 stars 200 forks source link

Open aicentral opened 6 years ago

aicentral commented 6 years ago

DataParallel should be used only with GPUs. If a user disabled Cuda in order to run the model on CPU, an error will show up.