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)
参考
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)
参考