# @package _global_
general:
    name : 'noh_uni3'
    gpus : 1
    wandb: 'online'
    test_only: null
    resume: null
    evaluate_all_checkpoints: False
train:
    n_epochs: 2000
    batch_size: 1024
    save_model: True
    num_workers: 4
    lr: 0.0001
model:
    n_layers: 10
    lambda_train: [5, 0.4, 1, 2, 0]
    nu:
        p: 1
        x: 1
        c: 1
        e: 1
        y: 1
dataset:
    remove_h: True
    pin_memory: True