U-NET Light: Training and Validation loss