| { | |
| "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 | |
| } | |
| } |