多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)