RuntimeError: size mismatch, m1: [32 x 768], m2: [2352 x 300] at C:\w\1\s\tmp_conda_3.7_104508\conda

RuntimeError: size mismatch, m1: [32 x 768], m2: [2352 x 300] at C:\w\1\s\tmp_conda_3.7_104508\conda\conda-bld\pytorch_1572950778684\work\aten\src\TH/generic/THTensorMath.cpp:197

原因:本人输入到model的第一层时,输入数据的第二维度值与model第一层的参数不一样
这里,本人输入数据维度为[32 x 768],但是model第一层是[2352 x 300]
因此,
方法:将model的[2352 x 300]改为768x300即可

你可能感兴趣的:(Python3,python)