深度学习代码损失,常会遇到如果发生什么条件,我们就返回一个预设值的情况,比如:在我的人体姿态估计网络中,如果该张图没有人体实例,我就不返回不更新梯度,损失为0.
但我直接返回int值 0 ,会报错,这个时候只需要将0更改为:
torch.tensor(0.0,requires_grad=True)
深度学习代码损失,常会遇到如果发生什么条件,我们就返回一个预设值的情况,比如:在我的人体姿态估计网络中,如果该张图没有人体实例,我就不返回不更新梯度,损失为0.
但我直接返回int值 0 ,会报错,这个时候只需要将0更改为:
torch.tensor(0.0,requires_grad=True)