Free used recycling tensors

326e08de · Gustaf Ahdritz · a7e631cc · 326e08de
Commit 326e08de authored Nov 18, 2021 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

openfold/model/model.py openfold/model/model.py +1 -1

No files found.
--- a/openfold/model/model.py
+++ b/openfold/model/model.py
@@ -233,7 +233,7 @@ class AlphaFold(nn.Module):
            z = z + z_prev_emb

            # Possibly prevents memory fragmentation 
-            del m_1_prev_emb, z_prev_emb
+            del m_1_prev, z_prev, x_prev, m_1_prev_emb, z_prev_emb

        # Embed the templates + merge with MSA/pair embeddings
        if self.config.template.enabled: