反复检查了,输出的size不可能是40000,所以为什么就是不行呢?最后改了一行代码,程序终于能正常跑了
x= x.reshape(225, -1)
改成:
x = nn.Flatten()(x)
这两行代码功能是差不多的。
代码能正常跑起来之后才知道,x的第一维不总是225,所以更稳妥的做法是直接用 Flatten
反复检查了,输出的size不可能是40000,所以为什么就是不行呢?最后改了一行代码,程序终于能正常跑了
x= x.reshape(225, -1)
改成:
x = nn.Flatten()(x)
这两行代码功能是差不多的。
代码能正常跑起来之后才知道,x的第一维不总是225,所以更稳妥的做法是直接用 Flatten