Fix assert bug

1d878a12 · Gustaf Ahdritz · GitHub · 7fdb503e · 1d878a12
Unverified Commit 1d878a12 authored Jul 07, 2023 by Gustaf Ahdritz Committed by GitHub Jul 07, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

train_openfold.py train_openfold.py +0 -1

No files found.
--- a/train_openfold.py
+++ b/train_openfold.py
@@ -88,7 +88,6 @@ class OpenFoldWrapper(pl.LightningModule):
            )

        for k,v in other_metrics.items():
-            assert(len(v.shape) == 1)
            self.log(
                f"{phase}/{k}",
                torch.mean(v),