code: def load_base_shapes(filename):
      """Get a dict of `InfShape` from a filename."""
      with open(filename, "r") as f:
          d = yaml.safe_load(f)
      return {k: InfShape.from_base_shape(v) for k, v in d.items()}
  
file_path: NeMo/nemo/collections/nlp/modules/common/megatron/mup/shape.py
imports: from copy import deepcopy
  import yaml
  from torch import nn
  from torch.nn import Linear
  from torch.nn.modules.conv import _ConvNd
  from nemo.collections.nlp.modules.common.megatron.mup.infshape import InfShape, zip_infshape
  from nemo.collections.nlp.modules.common.megatron.mup.layer import MuReadout, rescale_linear_bias
label_desc: Use image node_id 4 for calling a global function with example usage: load_base_shapes(filename) and returns: unknown