DataParallel layers (multi GPU)

import torch.nn as nn

import torch.nn.parallel

import torch.backends.cudnn as cudnn

model = nn.DataParallel(model)




阅读更多
个人分类: gpu
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

DataParallel layers (multi GPU)

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭