2018年11月5日
pytorch 多gpu训练
摘要： pytorch 多gpu训练用nn.DataParallel重新包装一下数据并行有三种情况前向过程device_ids=[0, 1, 2]model = model.cuda(device_ids[0])model = nn.DataParallel(mod... 阅读全文
posted @ 2018-11-05 10:59 overfitover 阅读(3409) 评论(0) 推荐(0) 编辑