1 model.load_state_dict(torch.load(modelpath))------> model.load_state_dict(torch.load(modelpath, map_location=torch.device('cpu'))) 2 将出现 ".cuda()"的地方全修改为“.to(device)”
本文介绍如何在PyTorch中正确地加载预训练模型，并针对不同硬件环境进行适配，包括使用CPU而非GPU加载模型的方法，以及统一使用“.to(device)”替代之前的“.cuda()”操作。
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
