跑代码的时候遇到了
ImportError: cannot import name ‘_LinearWithBias’ from ‘torch.nn.modules.linear’ 这个问题,
搜了好久都没找到
原因是pytorch版本问题
版本在torch>1.10之后就没有_LinearWithBias了
将代码改为
if float(torch.__version__.split('.')[0]) == 0 or (float(torch.__version__.split('.')[0]) == 1 and float(torch.__version__.split('.')[1])) < 9:
from torch.nn.modules.linear import _LinearWithBias
else:
from torch.nn.modules.linear import NonDynamicallyQuantizableLinear as _LinearWithBias
就导入了
版权声明:本文为m0_52494023原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。