| defaults: | |
| - _self_ | |
| - paths: default | |
| - hydra: default # ← tells Hydra to use the logging/output config | |
| - data_module: pair | |
| - model: classifier | |
| - trainer: gpu | |
| - extras: default | |
| - logger: wandb | |
| - callbacks: default | |
| # experiment configs allow for version control of specific hyperparameters | |
| # e.g. best hyperparameters for given model and datamodule | |
| - experiment: null | |
| # config for hyperparameter optimization | |
| - hparams_search: null | |
| # debugging config (enable through command line, e.g. `python train.py debug=default) | |
| - debug: null | |
| task_name: train/${model} | |
| # tags to help you identify your experiments | |
| # you can overwrite this in experiment configs | |
| # overwrite from command line with `python train.py tags="[first_tag, second_tag]"` | |
| tags: ["dev"] | |
| # set False to skip model training | |
| train: True | |
| # evaluate on test set, using best model weights achieved during training | |
| # lightning chooses best weights based on the metric specified in checkpoint callback | |
| test: True | |
| # simply provide checkpoint path to resume training | |
| ckpt_path: null | |
| # seed for random number generators in pytorch, numpy and python.random | |
| seed: 42 |