| # @package _global_ | |
| general: | |
| name : 'debug' | |
| gpus : 0 | |
| wandb: 'disabled' | |
| sample_every_val: 1 | |
| samples_to_generate: 4 | |
| samples_to_save: 2 | |
| chains_to_save: 1 | |
| remove_h: True | |
| number_chain_steps: 10 # Number of frames in each gif | |
| check_val_every_n_epochs: 1 | |
| train: | |
| batch_size: 4 | |
| save_model: False | |
| n_epochs: 2 | |
| model: | |
| n_layers: 2 | |
| hidden_mlp_dims: {'X': 17, 'E': 18, 'y': 19 } | |
| hidden_dims: {'dx': 20, 'de': 21, 'dy': 22, 'n_head': 5, 'dim_ffX': 23, 'dim_ffE': 24, 'dim_ffy': 25} | |
| sample: | |
| sample_steps: 10 | |