| # @package _global_ | |
| # to execute this experiment run: | |
| # python train.py experiment=example | |
| defaults: | |
| - override /datamodule: focus150.yaml | |
| - override /model: focus150.yaml | |
| - override /callbacks: default.yaml | |
| - override /logger: many_loggers | |
| - 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: "focusMSE_150" | |
| seed: 12345 | |
| trainer: | |
| min_epochs: 1 | |
| max_epochs: 100 | |
| model: | |
| _target_: src.models.focus_module.FocusMSELitModule | |
| lin1_size: 128 | |
| lin2_size: 256 | |
| lin3_size: 64 | |
| datamodule: | |
| batch_size: 128 | |