general:
    name: 'graph_dit'
    wandb: 'disabled' 
    gpus: 1
    gpu_number: 3
    resume: null
    test_only: null
    sample_every_val: 2500
    samples_to_generate: 512      
    samples_to_save: 3
    chains_to_save: 1
    log_every_steps: 50
    number_chain_steps: 8
    final_model_samples_to_generate: 100
    final_model_samples_to_save: 20
    final_model_chains_to_save: 1
    enable_progress_bar: False
    save_model: True
model:
    type: 'discrete'
    transition: 'marginal'                  
    model: 'graph_dit'
    diffusion_steps: 500
    diffusion_noise_schedule: 'cosine'
    guide_scale: 2
    hidden_size: 1152
    depth: 6
    num_heads: 16
    mlp_ratio: 4
    drop_condition: 0.01
    lambda_train: [1, 10]  # node and edge training weight 
    ensure_connected: True
train:
    n_epochs: 5000
    batch_size: 1200
    lr: 0.0002
    clip_grad: null
    num_workers: 0
    weight_decay: 0
    seed: 0
    val_check_interval: null
    check_val_every_n_epoch: 1
dataset:
    datadir: 'data/'
    task_name: 'nasbench-201'
    guidance_target: 'nasbench-201'
    pin_memory: False