| { | |
| "tasks": { | |
| "regression_task": true, | |
| "classification_task": null, | |
| "mclassification_task": null | |
| }, | |
| "optimizer": { | |
| "lrate": 0.0001, | |
| "weight_decay": 0.0001, | |
| "clip": 1, | |
| "betas": [ | |
| 0.9, | |
| 0.999 | |
| ], | |
| "eps": 1e-08, | |
| "schedule_lr": false, | |
| "min_lrate": 0, | |
| "warmup_iters": 0, | |
| "lr_decay_iters": 0, | |
| "amsgrad": false | |
| }, | |
| "params": { | |
| "mol_in_channels": 43, | |
| "prot_in_channels": 33, | |
| "prot_evo_channels": 1280, | |
| "hidden_channels": 200, | |
| "aggregators": [ | |
| "mean", | |
| "min", | |
| "max", | |
| "std" | |
| ], | |
| "scalers": [ | |
| "identity", | |
| "amplification", | |
| "linear" | |
| ], | |
| "pre_layers": 2, | |
| "post_layers": 1, | |
| "total_layer": 3, | |
| "K": [ | |
| 5, | |
| 10, | |
| 20 | |
| ], | |
| "dropout": 0, | |
| "dropout_attn_score": 0.2, | |
| "heads": 5 | |
| } | |
| } |