return "val_loss" in validation step and "avg_val�_loss" from validation_epoch_end in trainer 
I want to return  the val_loss from the validation step and avg_val_loss from validation epoch end in the trainer.