code: def model(c: MNISTConfigs):
      """
      ### Create model
  
      We use [`MNISTConfigs`](../../experiments/mnist.html#MNISTConfigs) configurations
      and set a new function to calculate the model.
      """
      return Model().to(c.device)
  
file_path: annotated_deep_learning_paper_implementations/labml_nn/normalization/batch_norm/mnist.py
imports: import torch.nn
  import torch.nn.functional
  import torch.utils.data
  from labml import experiment
  from labml.configs import option
  from labml_helpers.module import Module
  from labml_nn.experiments.mnist import MNISTConfigs
  from labml_nn.normalization.batch_norm import BatchNorm
label_desc: Use image node_id 1 for calling a global function with example usage: model(c) and returns: Model