model.to(device), model.cuda(), model.cpu(), DataParallel

Pytorch | Pytorch框架中模型和数据的gpu和cpu模式：

model.to(device), model.cuda(), model.cpu(), DataParallel

转载自：https://blog.csdn.net/iLOVEJohnny/article/details/106021547