lucidrains / alphafold2

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

the dimensions of coordinates in CASP12 #71

Closed zhangyi-taotao closed 3 years ago

zhangyi-taotao commented 3 years ago

@lucidrains Hi! In the test_coor, there is code like: assert coords.shape == (2, 16 14, 3), 'must output coordinates' . What does “14“ here mean。In the model, num_backbone_atoms = 3, shouldn’t the coordinate dimension be(2,16 3, 3)。

lucidrains commented 3 years ago

@zhangyi-taotao hey yes, that is a bug indeed and i'll get around to fixing it today at the very latest tomorrow!

lucidrains commented 3 years ago

@zhangyi-taotao https://github.com/lucidrains/alphafold2#atoms let me know if that helps or not!

zhangyi-taotao commented 3 years ago

@lucidrains Thank you very much for helping me answer my doubts!