robust Charbonnier loss function

robust Charbonnier loss function_第1张图片

这篇论文中有使用  Charbonnier loss

与L1相比,增加了一个正则项ε。用Charbonnier Loss来近似L1损失来提高模型的性能,接近零点的值的梯度由于ε的存在,梯度不会太小,避免梯度消失。

pytorch中Charbonnier loss的使用

robust Charbonnier loss function_第2张图片

 

在train.py中调用

就可以对网络进行训练了 

你可能感兴趣的:(人工智能)