# @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