发现在使用如下torch.nn.DataParallel,原有模型的字典结构会发生一定变化
model = Model()
if torch.cuda.device_count() > 1:
model = nn.DataParallel(model,device_ids=[0,1,2<
发现在使用如下torch.nn.DataParallel,原有模型的字典结构会发生一定变化
model = Model()
if torch.cuda.device_count() > 1:
model = nn.DataParallel(model,device_ids=[0,1,2<