File size: 1,732 Bytes
a2ffdeb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
{
  "affinity_model_args": {
    "esm_embed_dim": 1280,
    "pairformer_args": {
      "activation_checkpointing": true,
      "dropout": 0.1,
      "num_blocks": 8,
      "pairwise_head_width": 32,
      "pairwise_num_heads": 4,
      "post_layer_norm": false
    },
    "transformer_args": {
      "activation_checkpointing": false,
      "classification_head": true,
      "num_blocks": 2,
      "num_heads": 4,
      "token_s": 384
    }
  },
  "affinity_model_args2": {
    "esm_embed_dim": 1280,
    "pairformer_args": {
      "activation_checkpointing": true,
      "dropout": 0.1,
      "num_blocks": 8,
      "pairwise_head_width": 32,
      "pairwise_num_heads": 4,
      "post_layer_norm": false
    },
    "transformer_args": {
      "activation_checkpointing": false,
      "classification_head": true,
      "num_blocks": 2,
      "num_heads": 4,
      "token_s": 384
    }
  },
  "affinity_prediction": true,
  "atom_feature_dim": 390,
  "atom_s": 128,
  "atom_z": 16,
  "atoms_per_window_keys": 128,
  "atoms_per_window_queries": 32,
  "embedder_args": {
    "add_additional_atom_features": true,
    "atom_encoder_depth": 3,
    "atom_encoder_heads": 4,
    "pocket_conditioning": false
  },
  "esm_module_args": {
    "dropout": 0.05,
    "esm_embed_dim": 1280,
    "esm_num_layers": 34,
    "use_esm_all_layers": false
  },
  "max_dist": 22.0,
  "num_dist_bins": 64,
  "pairformer_model_args": {
    "activation_checkpointing": true,
    "dropout": 0.25,
    "num_blocks": 48,
    "pairwise_head_width": 32,
    "pairwise_num_heads": 4,
    "post_layer_norm": false
  },
  "predict_args": {
    "pose_protein_cutoff": 15.0,
    "recycling_steps": 5
  },
  "token_s": 384,
  "token_z": 128,
  "use_kernels": true
}