bindcraft2 / WEIGHTS_README.md
anindya64's picture
BindCraft2 checkpoints: AlphaFold2 params (2022-12-06) + ProteinMPNN neutral/negative/positive variants
c40205d verified
|
Raw History Blame Contribute Delete
1.42 kB
# Shipped model weights
`proteinmpnn/` holds the ProteinMPNN checkpoints BC2 redesigns with, so an installation carries them
and no campaign names a path. Three variants of the same four models, each directory named for the one it holds: `weights_neutral/`
is the original release, `weights_negative/` the soluble-protein retraining BC2 uses by default, `weights_positive/`
the HyperMPNN thermostability retraining, each named for the surface charge it favours.
`mpnn_variant` selects between them and `mpnn_model` between `v_48_002`, `v_48_010`, `v_48_020` and
`v_48_030`, the backbone-noise levels the models were trained at. Each checkpoint is an `.npz`
holding one array per parameter, keyed `<module>|<parameter>` beside the `num_edges` and
`noise_level` it was trained with, so loading one reads no pickle. Neutral and negative weights are
from `dauparas/ProteinMPNN` (MIT), positive weights from `meilerlab/HyperMPNN` (MIT).
`alphafold/` is empty in the repository and is where a container bakes its AlphaFold parameters. The
2022-12-06 release is 5.3 GB, so an ordinary installation downloads it once into
`~/.cache/bindcraft/alphafold` on its first campaign instead. `bindcraft fetch-weights` does the same
download on demand, which is what a compute node with no route to the internet needs run for it on a
login node first. The parameters are CC-BY-4.0, DeepMind, and their licence travels in the archive.