Skip to content

GitLab

Commit 9a8910f9 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Remove exception for num_recycle=0

parent e310bba4
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
openfold/model/model.py
View file @ 9a8910f9
...@@ -239,9 +239,10 @@ class AlphaFold(nn.Module): ...@@ -239,9 +239,10 @@ class AlphaFold(nn.Module):
# altogether. We zero them this way instead of computing them # altogether. We zero them this way instead of computing them
# conditionally to avoid leaving parameters unused, which has annoying # conditionally to avoid leaving parameters unused, which has annoying
# implications for DDP training. # implications for DDP training.
if(not _recycle): # EDIT: This has since been removed from the official codebase (2cd61a)
m_1_prev_emb *= 0 # if(not _recycle):
z_prev_emb *= 0 # m_1_prev_emb *= 0
# z_prev_emb *= 0
# [*, S_c, N, C_m] # [*, S_c, N, C_m]
m[..., 0, :, :] += m_1_prev_emb m[..., 0, :, :] += m_1_prev_emb
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment