ImportError: cannot import name ‘_LinearWithBias‘ from ‘torch.nn.modules.linear‘

  • Post author:
  • Post category:其他


跑代码的时候遇到了

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 版权协议,转载请附上原文出处链接和本声明。