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

跑代码的时候遇到了

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

就导入了

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
扎眼的阳光的头像扎眼的阳光普通用户
上一篇 2023年11月7日
下一篇 2023年11月7日

相关推荐