google-deepmind / alphafold

Open source code for AlphaFold.
Apache License 2.0
12.35k stars 2.21k forks source link

The proper way to find the maximum subbatch_size? #910

Open estrozi opened 6 months ago

estrozi commented 6 months ago

Hi,

I was trying to figure out the maximum value of subbatch_size parameter I could use with my GPU hoping that a too big subbatch_size would simply cause a OOM. However, it seems that if subbatch_size gets bigger than the sequence length some weird behaviors take place (like the number of recycling going crazy, ptm score become nan or stuff like that).

My question is: what's the proper way of finding the maximum subbatch_size for a given VRAMsize?

Thanks,

Leandro.