import torch.nn as nn
import torch.nn.parallel
import torch.backends.cudnn as cudnn
model = nn.DataParallel(model)
扫描二维码关注公众号，回复：
2275381 查看本文章
DataParallel layers (multi GPU)
猜你喜欢
转载自blog.csdn.net/doublechenchenchen/article/details/80346791
今日推荐
周排行