DataParallel、parallel.data_parallel

model_new = nn.DataParallel(model, device_ids)    返回一个新的model

output = nn.parallel.data_parallel(model, input, device_ids)    返回输出的数据

posted @ 2020-09-18 16:08  6+0  阅读(354)  评论(0)    收藏  举报