您当前的位置: 首页 >  pytorch

Better Bench

暂无认证

  • 1浏览

    0关注

    695博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

【Pytorch】解决Fan in and fan out can not be computed for tensor with fewer than 2 dimensions

Better Bench 发布时间:2021-03-17 22:07:10 ,浏览量:1

问题

使用TEXT.build_vocab时报错Fan in and fan out can not be computed for tensor with fewer than 2 dimensions

vectors.unk_init = init.xavier_uniform_ # 没有命中的token的初始化方式 TEXT.build_vocab(train, min_freq=5, vectors=vectors)

原因是torchtext版本太高,不支持一维的词向量,仅仅支持二维以上的

解决

(1)方法一 是torchtext的版本问题,更换torchtext版本为0.2.3 问题解决

pip install torchtext==0.2.3

(2)方法二 在torchtext的vocab.py中的Vectors类里修改一行源码

关注
打赏
1665674626
查看更多评论
立即登录/注册

微信扫码登录

0.0370s