File size: 403 Bytes
bbe7989
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
{
    "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
}