RuntimeError: Function AddmmBackward returned an invalid gradient at index 1 - got [20, 512] but exp
RuntimeError: Function AddmmBackward returned an invalid gradient at index 1 - got [20, 512] but expected shape compatible with [20, 4608]跟全连接层有关系。我报错的代码是total_loss.backward(),其中对应的网络最后有一个全连接层全连接层的代码是 nn.Linear(512, 1)我将上面的512改为报错的 nn.Linear(4608,