请问如何并行化训练神经网络模型?
发布网友
发布时间：2022-05-21 04:54
共1个回答
热心网友
时间：2023-10-15 03:05
我常用的是pytorch，可通过以下方法实现并行计算（单机多卡）：
new_net = nn.DataParallel(net, device_ids=[0, 1])
output = new_net(input)
通过device_ids参数可以指定在哪些GPU上进行优化
