model = torch.load(model_path, map_location='cpu')
没有cuda时,记得加上map_location='cpu',要不然会报错
map_location='cpu'