Pytorch lightning validation_epoch_end error

[Abhilash_Srivastava]

@nivesh_gadipudi

[x['val_loss'] for x in outputs]

Check if this part is returning the results as expected.