code: def instantiate_datamodule(self):
      super().instantiate_datamodule()
      # We need to register the datamodule's y_normalizer as sub-module
      # so that it gets moved to the current device.
      self.y_normalizer = self._datamodule.y_normalizer
  
imports: from typing import Any, List
  import inspect
  import torch
  import hydra
  from pytorch_lightning import LightningModule, LightningDataModule
  from torchmetrics import MetricCollection
  from einops import rearrange
  from omegaconf import OmegaConf
  from src.utils.utils import get_logger
  from src.optim.param_grouping import group_parameters_for_optimizer
  from src.utils.checkpoint import load_checkpoint
  import datasets
label_desc: Use image node_id 1 for calling the ModelwNormalizer obj's underlying member method code with example usage: obj.instantiate_datamodule() without return types