Skip to content

ERM Training

General Command:

Bash
python3 scripts/train_ERM.py --config-name CONFIG_NAME [options]

config layout:

yaml config
1
2
3
4
5
6
7
8
data:
  train_loader      # instantiatable
  val_loader: ???   # instantiatable (optional)
  test_loader: ???  # instantiatable (optional)

model           # instantiatable

trainer         # instantiatable

MNIST

Bash
python3 scripts/train_ERM.py --config-name mnist-mlp

CIFAR10/CIFAR100

CIFAR10

Bash
python3 scripts/train_ERM.py --config-name cifar10 [model=cifar_resnet50]

CIFAR100

Bash
python3 scripts/train_ERM.py --config-name cifar100 [model=cifar_resnet50]

ImageNet

Bash
python3 scripts/train_ERM.py --config-name imagenet-resnet50