同问
开发者问题收集
如何使用 ppc64le 和 x86 在不同版本的 pytorch(1.3.1 和 1.6.x)之间加载检查点?
正如我在此处概述的那样,由于硬件原因(例如使用 ppc64le IBM 架构),我只能使用旧版本的 pytorch 和 torchvision。因此,我在不同的计算机、集群和我的个人 Mac 之间发送和接收检查点时遇到问题。我想知道是否有任何方法可以加载模型以避免此问题?例如,在使用 1.6.x 时,也许可以使用旧格式和新格式保存模型。当然,对于 1.3.1 到 1.6.x 来说是不可能的,但至少
pytorch
ppc64le
2020-09-30
«
1
»