多GPU模型转换到cpu上

3. 多GPU模型转换到cpu上

通过DataParallel包装的model会再加一层module。所以state_dict会多一个module前缀。假设net1 是通过DataParallel包装的模型Net的实例，我们要把它装换到cpu上。方法就是重新建一个对象，把参数迁移过去

state_dict = net.module.state_dict()
net = Net()
net.load_state_dict(state_dict)

参考 

https://blog.csdn.net/u010510549/article/details/91390953

标签：模型,module,DataParallel,state,dict,GPU,net,cpu
来源： https://www.cnblogs.com/rabitvision/p/16134525.html