我代码中造成警告的语句是:文章来源:https://www.toymoban.com/news/detail-515278.html
value_loss = F.mse_loss(predicted_value, td_value) # predicted_value是预测值,td_value是目标值,用MSE函数计算误差
原因:mse_loss损失函数的两个输入Tensor的shape不一致。经过reshape或者一些矩阵运算以后使得shape一致,不再出现警告了。文章来源地址https://www.toymoban.com/news/detail-515278.html
到了这里,关于【Pytorch警告】Using a target size (torch.Size([])) that is different to the input size (torch.Size([1])的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!