Closed rarilurelo closed 5 years ago
torch.mean(loss * out_masks) should be replaced by torch.sum(loss * out_masks) / torch.sum(out_masks)
torch.mean(loss * out_masks)
torch.sum(loss * out_masks) / torch.sum(out_masks)
torch.mean(loss * out_masks)
should be replaced bytorch.sum(loss * out_masks) / torch.sum(out_masks)