Merge branch 'main' of ssh://github.com/aqlaboratory/openfold into main

0febc3e5 · Gustaf Ahdritz · 1c3d31bb · 30e051cf · 0febc3e5
Commit 0febc3e5 authored Jan 03, 2022 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

train_openfold.py train_openfold.py +2 -0

No files found.
--- a/train_openfold.py
+++ b/train_openfold.py
@@ -49,6 +49,8 @@ class OpenFoldWrapper(pl.LightningModule):
            model=self.model, decay=config.ema.decay
        )
        
+        self.cached_weights = None
+
    def forward(self, batch):
        return self.model(batch)