Free used recycling tensors

aqlaboratory · Nov 18, 2021 · 326e08d · 326e08d
1 parent a7e631c
commit 326e08d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openfold/model/model.py b/openfold/model/model.py
@@ -233,7 +233,7 @@ def iteration(self, feats, m_1_prev, z_prev, x_prev, _recycle=True):
             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: