vnet.pytorch icon indicating copy to clipboard operation
vnet.pytorch copied to clipboard
verified publisher icon mattmacy

fix a bug in CPU mode

Open aicentral opened this issue 7 years ago • 0 comments

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.

aicentral avatar Jul 06 '18 16:07 aicentral