模型load文件时报AttributeError: Can't get attribute 'Cours' on <module '__main__' from 错误

解决方法:

Pytorch使用Pickle来处理保存/加载模型,这个问题实际上是Pickle的问题,而不是Pytorch。

解决方法也非常简单,只需显式地导入类定义。即将包含类定义的文件复制粘贴到与要运行的文件同一文件夹下,再

import Class!

posted @ 2019-12-10 18:19  闪存第一菜鸡  阅读(8691)  评论(0编辑  收藏  举报