解决办法:在torch.load()之前,加入你定义的模型的CLASS,也就是把训练模型时的定义的模型的类复制到torch.load()之前。
解决pytouch导入模型torch.load报错:AttributeError: Can‘t get attribute ‘XXX‘ on <module ‘__main__‘ from XXX>
于 2024-01-02 17:02:46 首次发布
解决办法:在torch.load()之前,加入你定义的模型的CLASS,也就是把训练模型时的定义的模型的类复制到torch.load()之前。