I am getting error like “if input is not None and input.dim() != 4:
AttributeError: ‘tuple’ object has no attribute ‘dim’” when I try to return those two values.
When I won’t return pLoss and just return the calculated value. It won’t throw any error.