luizz8416

张量尺寸错误 RuntimeError: output with shape [1, 128, 128] doesn't match the broadcast shape [3, 128, 128]

8

完整的报错信息 Train Epoch: 0 [0/394] D_loss: 1.4287535548210144 = DisF:0.9066857099533081 + DisT:0.5220678448677063 En_De_loss: 0.6271777153015137 = Gen:0.5223720073699951 + En:1.126967191696167=MSE-0.49768513441085815+MSSIM-125.85639953613281, De:0.4012618660926819 Traceback (most recent call last): File "train_ed.py", line 90, in for batch_idx,...