Remove removed function call

ceb95303 · Gustaf Ahdritz · 82e02065 · ceb95303
Commit ceb95303 authored Jun 14, 2022 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

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

No files found.
--- a/openfold/model/model.py
+++ b/openfold/model/model.py
@@ -443,7 +443,6 @@ class AlphaFold(nn.Module):
            is_final_iter = cycle_no == (num_iters - 1)
            with torch.set_grad_enabled(is_grad_enabled and is_final_iter):
                if is_final_iter:
-                    self._enable_activation_checkpointing()
                    # Sidestep AMP bug (PyTorch issue #65766)
                    if torch.is_autocast_enabled():
                        torch.clear_autocast_cache()