跑代码的时候遇到了
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
就导入了
文章出处登录后可见!
已经登录?立即刷新