D:\py\anaconda\envs\pytorch\lib\site-packages\torch\nn\modules\loss.py:536: UserWarning: Using a target size (torch.Size([1, 200, 1])) that is different to the input size (torch.Size([200, 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-10-08 15:05:22 浏览: 99
python 3.7 caffe 解压到 D:\Anaconda3\envs\caffe\Lib
这是一个PyTorch的警告信息,提示你输入的大小(shape)与目标(target)的大小不同,可能会导致错误的结果。具体来说,输入的大小为[200, 1],目标的大小为[1, 200, 1],这意味着目标在第0维上有一个额外的维度。这可能是由于数据预处理过程中的错误或者数据维度的混淆导致的。你需要检查一下你的代码,确保输入和目标的大小是对应的,并且没有混淆维度。如果你确定大小是正确的,你可以通过调用`target.squeeze()`方法来移除目标张量中的额外维度。
阅读全文