pytorch使用多GPU进行训练batch加载的数据只有1/GPUS

使用 torch.nn.DataParallel 进行多卡训练是,debug时加载的数据只有1/GPUS

Guess you like

Origin blog.csdn.net/AAliuxiaolei/article/details/120128604