About the recycle dimension

hpcaitech / FastFold

Optimizing AlphaFold Training and Inference on GPU Clusters

Apache License 2.0

556 stars 84 forks source link

About the recycle dimension #169

Open semal opened 1 year ago

semal commented 1 year ago

I found that an additional dimension representing the number of iterations was added to the input data. I'm wondering what is the purpose of this dimension? Will all model parameters also increase in dimension during training? I did not see this kind of data processing in the original JAX version of AlphaFold.

Gy-Lu commented 1 year ago

Hi, this part follows the AlphaFold2 paper. See 1.10 of Supplementary information of AF2.