frozenleaves/alphafold3
1
0
Fork 0
mirror of https://github.com/google-deepmind/alphafold3.git synced 2025-10-20 13:23:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix flag descriptions for save_embeddings and save_distogram

Browse Source 
PiperOrigin-RevId: 758180480
Change-Id: Iaccb91b1f0be4a6ea6d4b9b77c55d5593a4afd03
This commit is contained in:
Augustin Zidek
2025-05-13 05:51:52 -07:00
committed by Copybara-Service
parent 267ab05055
commit 38a9333b54
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
run_alphafold.py
View File

@ -272,13 +272,15 @@ _NUM_SEEDS = flags.DEFINE_integer(
_SAVE_EMBEDDINGS = flags.DEFINE_bool(
'save_embeddings',
False,
'Whether to save the final trunk single and pair embeddings in the output.',
'Whether to save the final trunk single and pair embeddings in the output.'
' Note that the embeddings are large float16 arrays: num_tokens * 384'
' + num_tokens * num_tokens * 128.',
)
_SAVE_DISTOGRAM = flags.DEFINE_bool(
'save_distogram',
False,
'Whether to save the final distogram in the output. Note that the distoram '
'large: num_tokens * num_tokens * 39.',
'Whether to save the final distogram in the output. Note that the distogram'
' is a large float16 array: num_tokens * num_tokens * 64.',
)
_FORCE_OUTPUT_DIR = flags.DEFINE_bool(
'force_output_dir',