U-NET: Training and Validation loss