其他分享
首页 > 其他分享> > 多GPU模型转换到cpu上

多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