作者:眼睛Blank | 来源:互联网 | 2023-06-07 13:07
原因:主要还是pytorch版本问题。高版本的torch.save保存模型时,使用的新的格式,而torch.load使用的是旧格式。不知道为啥,我在同一个环境下,先save后load也会出这个错。*
原因:主要还是pytorch
版本问题。高版本的torch.save
保存模型时,使用的新的格式,而torch.load
使用的是旧格式。不知道为啥,我在同一个环境下,先save后load也会出这个错。
**解决:**加一个参数