我正在做一个项目,我最初训练模型并存储最佳模型。为了评估经过训练的模型,我尝试使用 model.load_state_dict(torch.load(path)),其中路径指定检查点。当我在 gpu 机器上运行该模型时,该模型没有给出一致的结果,但代码在 cpu 的本地机器上运行良好。以前有人遇到过类似的问题吗?
我正在做一个项目,我最初训练模型并存储最佳模型。为了评估经过训练的模型,我尝试使用 model.load_state_dict(torch.load(path)),其中路径指定检查点。当我在 gpu 机器上运行该模型时,该模型没有给出一致的结果,但代码在 cpu 的本地机器上运行良好。以前有人遇到过类似的问题吗?