TensorBoard logging in validation_step and test_step
Even defining the log in all steps of the PL model:
    def training_step(self, batch, batch_idx):
        ...
        # TensorBoard logging
        log = {"train_loss": train_loss, "train_mrr": train_mrr}
        return {'loss': train_loss, "log":log}

    def test_step(self, batch, batch_idx):
        ...
        # TensorBoard logging
        log = {"test_loss": test_loss, "test_mrr": test_mrr}
        return {'test_loss': test_loss, 'log': log}

    def validation_step(self, batch, batch_idx):
        ...
        # TensorBoard logging
        log = {"val_loss": val_loss, "val_mrr": val_mrr}
        return {'val_loss': val_loss, 'log': log}
I could visualize only the train_loss and train_mrr on TensorBoard. I mean, test_loss and test_mrr do not appear in TesorBoard. Similar behavior happens with val_loss and val_mrr.