Remove exception for num_recycle=0

9a8910f9 · Gustaf Ahdritz · e310bba4 · 9a8910f9
Commit 9a8910f9 authored Apr 13, 2022 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

openfold/model/model.py openfold/model/model.py +4 -3

No files found.
--- a/openfold/model/model.py
+++ b/openfold/model/model.py
@@ -239,9 +239,10 @@ class AlphaFold(nn.Module):
        # altogether. We zero them this way instead of computing them
        # conditionally to avoid leaving parameters unused, which has annoying
        # implications for DDP training.
-        if(not _recycle):
-            m_1_prev_emb *= 0
-            z_prev_emb *= 0
+        # EDIT: This has since been removed from the official codebase (2cd61a)
+#        if(not _recycle):
+#            m_1_prev_emb *= 0
+#            z_prev_emb *= 0

        # [*, S_c, N, C_m]
        m[..., 0, :, :] += m_1_prev_emb