{
  "train": {
    "seed": 516,
    "n_epochs": 2000,
    "n_report_steps": 1000,
    "analyze_epochs": 1,
    "validate_epochs": 1,
    "grad_accumulate": 1,
    "n_samples": 64
  },
  "optimization": {
    "clip_grad": true,
    "lr": 0.0003,
    "resume": false,
    "weight_decay": 1e-12,
    "factor": 0.8,
    "patience": 4,
    "min_lr": 1e-05,
    "ema_decay": 0.9999
  },
  "dynamics": {
    "tanh": false,
    "attention": false,
    "beta1": 3.0,
    "atom_order": 2,
    "edge_order": 3,
    "hidden_dim": 256,
    "n_layers": 9,
    "atom_dim": 13,
    "edge_dim": 5,
    "t_dim": 1,
    "name": "egnn_b",
    "sample_steps": 20,
    "sigma1_pos": 0.005,
    "t_min": 0.0001,
    "w_edge": 10,
    "error_warmup_edge": 1.5
  },
  "dataset": {
    "name": "qm9_drugs",
    "smiles_file": "./data/qm9/smiles.pkl",
    "train_data": "./data/qm9/train_test_data.pkl",
    "valid_data": "./data/qm9/val_test_data.pkl",
    "batch_size": 64,
    "num_workers": 4,
    "remove_h": false
  }
}