libokj's picture
Upload 154 files
416833a
raw
history blame
177 Bytes
_target_: deepscreen.models.predictors.transformer_cpi.TransformerCPI
protein_dim: 100
atom_dim: 34
hid_dim: 64
n_layers: 3
n_heads: 8
pf_dim: 256
dropout: 0.1
kernel_size: 7