code: def cifar10_train_augmented():
      """
      ### Augmented CIFAR 10 train dataset
      """
      from torchvision.datasets import CIFAR10
      from torchvision.transforms import transforms
  
      return CIFAR10(
          str(lab.get_data_path()),
          train=True,
          download=True,
          transform=transforms.Compose(
              [
                  # Pad and crop
                  transforms.RandomCrop(32, padding=4),
                  # Random horizontal flip
                  transforms.RandomHorizontalFlip(),
                  #
                  transforms.ToTensor(),
                  transforms.Normalize(
                      (0.5, 0.5, 0.5), (0.5, 0.5, 0.5)
                  ),
              ]
          ),
      )
  
file_path: annotated_deep_learning_paper_implementations/labml_nn/experiments/cifar10.py
imports: from typing import List
  import torch.nn
  from labml import lab
  from labml.configs import option
  from labml_helpers.datasets.cifar10 import CIFAR10Configs
  from labml_helpers.module import Module
  from labml_nn.experiments.mnist import MNISTConfigs
label_desc: Use image node_id 1 for calling a global function with example usage: cifar10_train_augmented() and returns: CIFAR10