PyTorch用CPU加载GPU模型
#实例化模型
network = CNNModel()
#加载模型
model_path = "GPU模型"
network.load_state_dict(torch.load(model_path, map_location = torch.device('cpu')))
network.eval()

#实例化模型
network = CNNModel()
#加载模型
model_path = "GPU模型"
network.load_state_dict(torch.load(model_path, map_location = torch.device('cpu')))
network.eval()