search

lucidrains / alphafold2

To eventually become an unofficial Pytorch implementation / replication of Alphafold2, as details of the architecture get released
MIT License
1.54k stars 254 forks source link

[todo] checkpointing #81

Open lucidrains opened 3 years ago

lucidrains commented 3 years ago

make sure checkpointing works, or see if evoformer can be made reversible

lucidrains commented 3 years ago

welcoming contributions!

hypnopump commented 3 years ago

not my specialty ;(

lucidrains commented 3 years ago

https://github.com/lucidrains/alphafold2/commit/597762beedef6ecbf24f630ad62acbd76050f7d4 initial checkpointing for entire evoformers (main trunk and extra msa)