| { | |
| "atom_hidden_size": 32, | |
| "block_hidden_size": 32, | |
| "edge_size": 32, | |
| "n_layers": 4, | |
| "dropout": 0.0, | |
| "k_neighbors": 8, | |
| "global_message_passing": true, | |
| "bottom_global_message_passing": false, | |
| "fragmentation_method": "PS_300", | |
| "model_type": "ClassifierModel", | |
| "num_pred_layers": 3, | |
| "nonlinearity": "gelu", | |
| "pred_dropout": 0.0, | |
| "pred_hidden_size": 64 | |
| } |