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, 1)