只有cpu的时候加载模型 checkpoint = torch.load(model_path, map_location=torch.device(‘cpu’))
京公网安备 11010502049817号