DataParallel layers (multi GPU)

import torch.nn as nn

import torch.nn.parallel

import torch.backends.cudnn as cudnn

model = nn.DataParallel(model)




阅读更多
想对作者说点什么?

博主推荐

换一批

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