When logging my validation loss inside validation_step() in PyTorch Lighnting like this:
validation_step()
def validation_step(self, batch: Tuple[Tensor, Tensor], _bat