
原因:主要还是pytorch版本问题。高版本的torch.save保存模型时,使用的新的格式,而torch.load使用的是旧格式。不知道为啥,我在同一个环境下,先save后load也会出这个错。
**解决:**加一个参数
版权声明:本文为weixin_40802676原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。

原因:主要还是pytorch版本问题。高版本的torch.save保存模型时,使用的新的格式,而torch.load使用的是旧格式。不知道为啥,我在同一个环境下,先save后load也会出这个错。
**解决:**加一个参数