📜 RuntimeError: size mismatch, m1: [1536 x 3], m2: [4096 x 101] at opt conda conda-bld pytorch_1587428398394 work aten src TH generic THTensorMath.cpp:41 - 无论代码示例
📅 最后修改于: 2022-03-11 14:59:48.088000 🧑 作者: Mango
代码示例2
img = img.view(1, -1) # converts [3 x 256 x 256] to 1 x 196608
output = model(img)