diff options
Diffstat (limited to 'runs/ogbg-molesol_transformer_classic_T0_ns1_h128_e1_s0.json')
| -rw-r--r-- | runs/ogbg-molesol_transformer_classic_T0_ns1_h128_e1_s0.json | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/runs/ogbg-molesol_transformer_classic_T0_ns1_h128_e1_s0.json b/runs/ogbg-molesol_transformer_classic_T0_ns1_h128_e1_s0.json new file mode 100644 index 0000000..25d89f4 --- /dev/null +++ b/runs/ogbg-molesol_transformer_classic_T0_ns1_h128_e1_s0.json @@ -0,0 +1,37 @@ +{ + "dataset": "ogbg-molesol", + "tag": "ogbg-molesol_transformer_classic_T0_ns1_h128_e1_s0", + "view": "transformer", + "compute": "classic", + "T": 0, + "n_sup": 1, + "hidden": 128, + "agg_layers": 5, + "compute_layers": 2, + "epochs": 1, + "eval_every": 10, + "lr": 0.001, + "bs": 128, + "lam_q": 1.0, + "halt_max_steps": 8, + "halt_min_steps": 1, + "halt_target": "loss", + "halt_loss_threshold": 0.2, + "halt_exploration_prob": 0.1, + "q_warmup_epochs": 0, + "ema": 0.0, + "seed": 0, + "num_workers": 0, + "max_train_batches": 1, + "max_eval_batches": 1, + "metric": "rmse", + "sec": 0.8, + "dev": "cuda", + "ep": 1, + "val": { + "rmse": 4.1780009269714355 + }, + "test": { + "rmse": 4.270694255828857 + } +}
\ No newline at end of file |
