为什么从ubuntu到Windows的训练模型都出错?

时间:2019-08-07 07:01:51

标签: python-3.x pytorch

我已经在超级计算机(ubuntu)上训练了模型。训练后,我在Windows 10上使用了该模型,并出现了以下错误:

SourceChangeWarning: source code of class 'torch.nn.modules.linear.Linear' has changed. you can retrieve the original source code by accessing the object's source attribute or set `torch.nn.Module.dump_patches = True` and use the patch tool to revert the changes.
  warnings.warn(msg, SourceChangeWarning)

我无法加载我训练的模型。

ubuntu中的pytorch版本:1.1.0a0+9a7bcac

Windows中的

pytorch版本:0.4.1

出了什么问题,我该如何解决?

2 个答案:

答案 0 :(得分:0)

评论讨论后进行编辑:
您的Windows版本似乎已过时,因此发生版本冲突。我强烈建议将Windows上的版本更新为1.0版,以解决此问题。

根据this link,只要您的模型似乎仍然可以正常工作,就可以忽略警告(不是错误)。造成此类更改的常见原因是两个系统上的PyTorch版本不一致,因此可能会遇到此警告。

通常,这些版本应该是完全向后兼容的,但是当然不能为此提供保证。除非源代码检测到换行符的变化(这是我记得的主要区别),否则与您在Linux和/或Windows上的事实无关,尽管我认为这不太可能情况。

答案 1 :(得分:0)

非常感谢dennlinger! 是版本问题。

我更新了pytorch版本后,就可以了!