AttributeError: Can‘t get attribute ‘NonDynamicallyQuantizableLinear‘

原因:保存下来的模型和参数不能在没有类定义时直接使用。

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

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

最直接有效:在要运行的文件上复制类定义

所以找到提示的文件,加上这个类的代码

class NonDynamicallyQuantizableLinear(Linear):
    def __init__(self, in_features: int, out_features: int, bias: bool = True,
                 device=None, dtype=None) -> None:
        super().__init__(in_features, out_features, bias=bias,
                         device=device, dtype=dtype)

你可能感兴趣的:(深度学习,pytorch,python)