出现了类似的错误:
RuntimeError: size mismatch, m1: [64 x 1008], m2: [1024 x 64] at …\aten\src\TH/generic/THTensorMath.cpp:41
找不到解决办法,先别慌,看看自己的网络模型中的输入参数是不是输错了
报错展示:
错误展示:
把63,改为正确的64,就好了
RuntimeError: size mismatch, m1: [64 x 1008], m2: [1024 x 64] at ..\aten\src\TH/generic/THTensorMath
最新推荐文章于 2024-02-11 17:05:26 发布