from torch.nn import DataParallel
from torch.nn import DataParallerclass 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 ·
1357 阅读 ·
0 评论