exp_example_simple.yaml

# @package _global_

# to execute this experiment run:
# python run.py +experiment=exp_example_simple

defaults:
    - override /trainer: default_trainer.yaml           # choose trainer from 'configs/trainer/' folder or set to null
    - override /model: mnist_model.yaml                 # choose model from 'configs/model/' folder or set to null
    - override /optimizer: adam.yaml                 # choose model from 'configs/model/' folder or set to null
    - override /datamodule: mnist_datamodule.yaml       # choose datamodule from 'configs/datamodule/' folder or set to null
    - override /callbacks: default_callbacks.yaml       # choose callback set from 'configs/callbacks/' folder or set to null
    - override /logger: null                            # choose logger from 'configs/logger/' folder or set to null

# all parameters below will be merged with parameters from default configurations set above
# this allows you to overwrite only specified parameters

seed: 12345

trainer:
    min_epochs: 1
    max_epochs: 10
    gradient_clip_val: 0.5

model:
    lin1_size: 128
    lin2_size: 256
    lin3_size: 64

optimizer:
    lr: 0.001

datamodule:
    batch_size: 64
    train_val_test_split: [55_000, 5_000, 10_000]