Good day,
Thank you for the nice job and for making it open-source ~
I have some doubts about these two losses. Could you please explain why you have taken the "mean of L2 loss and TV loss"?
|
l2_loss = torch.norm(input[0], dim=1, p=2).mean() |
and
|
loss_var_l1 = (diff1.abs() / 255.0).mean() + (diff2.abs() / 255.0).mean() + ( |
Thank you ~
