| # @package _global_ | |
| # to execute this experiment run: | |
| # python train.py experiment=example | |
| defaults: | |
| - override /datamodule: mnist.yaml | |
| - override /model: mnist.yaml | |
| - override /callbacks: default.yaml | |
| - override /logger: null | |
| - override /trainer: default.yaml | |
| # all parameters below will be merged with parameters from default configurations set above | |
| # this allows you to overwrite only specified parameters | |
| # name of the run determines folder name in logs | |
| name: "simple_dense_net" | |
| seed: 12345 | |
| trainer: | |
| min_epochs: 10 | |
| max_epochs: 10 | |
| gradient_clip_val: 0.5 | |
| model: | |
| lin1_size: 128 | |
| lin2_size: 256 | |
| lin3_size: 64 | |
| lr: 0.002 | |
| datamodule: | |
| batch_size: 64 | |
| logger: | |
| wandb: | |
| tags: ["mnist", "${name}"] | |