美文网首页
Pytorch常遇到的错误汇总

Pytorch常遇到的错误汇总

作者: 吐舌小狗 | 来源:发表于2018-03-13 16:33 被阅读182次

    1.torch.cat时dim设置错误

    torch.cat((a,b),1) RuntimeError: inconsistent tensor sizes at /opt/conda/conda-bld/pytorch_1512378360668/work/torch/lib/TH/generic/THTensorMath.c:2864
    解决方案:查看变量的size,修改torch.dim中dim的值

    相关文章

      网友评论

          本文标题:Pytorch常遇到的错误汇总

          本文链接:https://www.haomeiwen.com/subject/imizfftx.html