(pytorch) sdy@swai:~/hw/pop2/CNN$ /home/sdy/anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/modules/loss.py:520: UserWarning: Using a target size (torch.Size([1, 384, 320])) that is different to the input size (torch.Size([384, 320])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size. return F.mse_loss(input, target, reduction=self.reduction)

这个警告信息表明你正在使用 PyTorch 中的均方误差损失函数 `F.mse_loss`，但输入和目标的大小不一致。这可能导致广播（broadcasting）操作，从而产生不正确的结果。建议确保输入和目标具有相同的大小。

根据警告信息，输入的大小为 `torch.Size([384, 320])`，而目标的大小为 `torch.Size([1, 384, 320])`。可以看到目标的第一个维度是1，而输入没有这个额外的维度。这可能是导致警告的原因。

为了解决这个问题，你可以调整目标的形状，使其与输入具有相同的形状。可以使用 `torch.squeeze()` 函数来去除目标张量中的尺寸为1的维度，或者使用 `torch.unsqueeze()` 函数在输入张量中添加一个尺寸为1的维度。

例如，如果你想要将目标形状调整为与输入相同的形状，可以使用以下代码：

target = target.squeeze()

或者使用以下代码：

target = target.unsqueeze(0)

请根据你的需求选择适当的方法，并确保输入和目标具有相同的形状，以避免广播操作导致的不正确结果。