config.yaml

seed: 42

dataset:
    name: MNIST
    root: .
    test_size: 0.2
    random_state: 42

dataloader:
    batch_size: 4

trainer:
    profiler: False
    gpus: 0
    fast_dev_run: False
    min_epochs: 1
    max_epochs: 1

logger:
    experiment_name: "mlflow-pipeline-MNIST"

optimizer:
    args:
      lr: 0.001
      weight_decay: 0.00001
    
    scheduler:
        enable:
        T_max: 5
        eta_min: 0.0001

num_class: 10


optuna:
    enable: True
    n_trials: 3
    parameters:
        optimizer:
          args:
            lr:
              type: float
              args:
                low: 0.01
                high: 0.03

    create_study:
        direction: minimize
        study_name: optuna
        storage: sqlite:///db/optuna.db