DataParallel layers (multi GPU)

import torch.nn as nn

import torch.nn.parallel

import torch.backends.cudnn as cudnn

model = nn.DataParallel(model)




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

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

关闭
关闭
关闭