code: def _small_student_model(c: Configs):
      """
      ### Create small model
      """
      return SmallModel().to(c.device)
  
file_path: annotated_deep_learning_paper_implementations/labml_nn/distillation/__init__.py
imports: import torch
  import torch.nn.functional
  from torch import nn
  from labml import experiment, tracker
  from labml.configs import option
  from labml_helpers.train_valid import BatchIndex
  from labml_nn.distillation.large import LargeModel
  from labml_nn.distillation.small import SmallModel
  from labml_nn.experiments.cifar10 import CIFAR10Configs
label_desc: Use image node_id 2 for calling a global function with example usage: _small_student_model(c) and returns: SmallModel