super-gradients/src/super_gradients/recipes/cifar10_resnet.yaml

# Cifar10 Classification Training:
# Reaches ~94.9 Accuracy after 250 Epochs
# Instructions:
#   0. Make sure that the data is stored in dataset_params.dataset_dir or add "dataset_params.data_dir=<PATH-TO-DATASET>" at the end of the command below (feel free to check ReadMe)
#   1. Move to the project root (where you will find the ReadMe and src folder)
#   2. Run the command:
#       python src/super_gradients/examples/train_from_recipe_example/train_from_recipe.py --config-name=cifar10_resnet +experiment_name=cifar10

defaults:
  - training_hyperparams: cifar10_resnet_train_params
  - dataset_params: cifar10_dataset_params
  - arch_params: resnet18_cifar_arch_params
  - checkpoint_params: default_checkpoint_params

train_dataloader: cifar10_train
val_dataloader: cifar10_val

data_loader_num_workers: 8

resume: False
training_hyperparams:
  resume: ${resume}


model_checkpoints_location: local
ckpt_root_dir:

architecture: resnet18_cifar

experiment_name: resnet18_cifar