code: def on_validation_epoch_end(self) -> None:
      if self.do_log_logits:
          logits = torch.cat(self._val_logits).detach().cpu()
          self._val_logits = []
          fig = self.create_plotly_histogram(logits, "logits")
          wandb.log(
              {
                  "valid_logits": wandb.Plotly(fig),
                  "global_step": self.global_step,
              },
              commit=False,
          )
      super().on_validation_epoch_end()
  
imports: import importlib
  import warnings
  from abc import ABCMeta, abstractmethod
  from functools import partial
  from typing import Any, Callable, Dict, List, Optional, Tuple, Type, Union
  import numpy
  import pytorch_lightning
  import torch
  import torch.nn
  import torchmetrics
  from omegaconf import DictConfig, OmegaConf
  from pandas import DataFrame
  from torch import Tensor
  from torch.optim import Optimizer
  from pytorch_tabular.models.common.heads import blocks
  from pytorch_tabular.models.common.layers import PreEncoded1dLayer
  from pytorch_tabular.utils import get_logger, reset_all_weights
label_desc: Use image node_id 28 for calling the BaseModel obj's underlying member method code with example usage: obj.on_validation_epoch_end() without return types