from torch.nn import DataParallel
from torch.nn import DataParaller
class torch.nn.DataParallel(module, device_ids=None, output_device=None, dim=0)
Parameters:
module – module to be parallelized
device_ids – CUDA devices (d...
原创
2019-03-18 21:26:37 ·
1368 阅读 ·
0 评论