google-deepmind / google-deepmind/alphafold

The proper way to find the maximum subbatch_size?

Open
#910 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
14.9k
Forks
2.9k
PR merge metrics
No merged PRs in 30d

Description

###
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.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.