D:\minicinda3\lib\site-packages\torch\nn\modules\loss.py:536: UserWarning: Using a target size (torch.Size([32])) that is different to the input size (torch.Size([512, 1])). 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)
时间: 2023-06-12 13:02:51 浏览: 279
Android.bp编译提示ninja: error: unknown target ‘MODULES-IN-xxx’终极指南
3星 · 编辑精心推荐
这个警告是因为你在使用 PyTorch 的 MSE Loss 时,输入的大小与目标的大小不匹配,导致 PyTorch 自动进行了广播,可能会造成错误的结果。请确保输入和目标的大小相同。你可以使用 `torch.reshape()` 或 `torch.squeeze()` 等函数调整输入和目标的大小,使它们匹配。另外,你也可以将 `reduction` 参数设置为 `'mean'` 或 `'sum'` 来避免该警告。
阅读全文