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

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

おすすめ

転載: blog.csdn.net/AAliuxiaolei/article/details/120128604