contextual_loss_pytorch
contextual_loss_pytorch copied to clipboard
Fixed some bugs about parameters and updated comments
I found some bugs and fixed it. Please check my pull request.
- Parameter
loss_type
is being ignored in bothContextualLoss
andContextualBilateralLoss
. - Parameter
band_width
inContextualBilateralLoss
is treated asweight_sp
. - Parameter
loss_type
is better to compared in lower case. - Default value of parameter
band_width
inContextualBilateralLoss
is0.5
butF.contextual_bilateral_loss
have default value of1.0
.
Thank you.