parkererickson/LGGM-Text2Graph

1hydra:2  job:3    chdir: True4  run:5    dir: ../outputs/${general.name}6general:7    name: 'cc_high'8    wandb: 'online'                # online | offline | disabled9    gpus: 110    resume: null11    test_only: null12    sample_every_val: 413    check_val_every_n_epochs: 1014    samples_to_generate: 100      15    samples_to_save: 316    chains_to_save: 117    log_every_steps: 5018    number_chain_steps: 819    final_model_samples_to_generate: 10020    final_model_samples_to_save: 3021    final_model_chains_to_save: 2022    condition: ' '23    setting: 'train_scratch'24    ckpt_path: null25model:26    type: 'discrete'27    transition: 'marginal'                  28    model: 'graph_tf'29    diffusion_steps: 50030    diffusion_noise_schedule: 'cosine'31    n_layers: 532    extra_features: 'all'33    hidden_mlp_dims: {'X': 256, 'E': 128, 'y': 128}34    hidden_dims: {'dx': 256, 'de': 64, 'dy': 64, 'n_head': 8, 'dim_ffX': 256, 'dim_ffE': 128, 'dim_ffy': 128}35    lambda_train: [5, 0]36train:37    n_epochs: 30038    batch_size: 839    accumulate_grad_batches: 140    lr: 0.000241    clip_grad: null42    save_model: True43    num_workers: 044    ema_decay: 045    weight_decay: 1e-1246    seed: 047    progress_bar: false48    optimizer: adamw49dataset:50    datadir: 'graph/'51    name: cc_high52    remove_h: null53    sample: 'seed'54