pytorch torch.load报错

model = torch.load(model_path, map_location='cpu')

没有cuda时,记得加上map_location='cpu',要不然会报错

猜你喜欢

转载自blog.csdn.net/qq_43523725/article/details/104302346