code: def main():
      """
      Run the experiment
      """
      experiment.create(name="capsule_network_mnist")
      conf = Configs()
      experiment.configs(
          conf,
          {
              "optimizer.optimizer": "Adam",
              "optimizer.learning_rate": 1e-3,
          },
      )
  
      experiment.add_pytorch_models({"model": conf.model})
  
      with experiment.start():
          conf.run()
  
imports: from typing import Any
  import torch.nn
  import torch.nn.functional
  import torch.utils.data
  from labml import experiment, tracker
  from labml.configs import option
  from labml_helpers.datasets.mnist import MNISTConfigs
  from labml_helpers.metrics.accuracy import AccuracyDirect
  from labml_helpers.module import Module
  from labml_helpers.train_valid import SimpleTrainValidConfigs, BatchIndex
  from labml_nn.capsule_networks import Squash, Router, MarginLoss
label_desc: Use image node_id 2 for calling a global function with example usage: main() without return types