is there a wrong in loss?

[xiaoxiaowannote]

may be your torch.lt is wrong

[huntkao]

asym_loss = torch.mean(if_asym * torch.abs(0.3 - torch.lt(gt_noise, est_noise).float()) * torch.pow(est_noise - gt_noise, 2)) should be asym_loss = torch.mean(if_asym * torch.abs(0.3 - torch.lt(est_noise, gt_noise).float()) * torch.pow(est_noise - gt_noise, 2)) ??