    def validation_epoch_end(self, outputs):
        """
        Computes average validation accuracy

        :param outputs: outputs after every epoch end

        :return: output - average valid loss
        """
        avg_loss = torch.stack([x["val_step_loss"] for x in outputs]).mean()
        self.log("val_loss", avg_loss, sync_dist=True)
Computes average validation accuracy

:param outputs: outputs after every epoch end

:return: output - average valid loss
tests/integration/frameworks/mlflow/MNIST/mnist_autolog_example.py