code: def main():
      # Create experiment
      experiment.create(name="ConvMixer", comment="cifar10")
      # Create configurations
      conf = Configs()
      # Load configurations
      experiment.configs(
          conf,
          {
              # Optimizer
              "optimizer.optimizer": "Adam",
              "optimizer.learning_rate": 2.5e-4,
              # Training epochs and batch size
              "epochs": 150,
              "train_batch_size": 64,
              # Simple image augmentations
              "train_dataset": "cifar10_train_augmented",
              # Do not augment images for validation
              "valid_dataset": "cifar10_valid_no_augment",
          },
      )
      # Set model for saving/loading
      experiment.add_pytorch_models({"model": conf.model})
      # Start the experiment and run the training loop
      with experiment.start():
          conf.run()
  
imports: from labml import experiment
  from labml.configs import option
  from labml_nn.experiments.cifar10 import CIFAR10Configs
label_desc: Use image node_id 2 for calling a global function with example usage: main() without return types