Fix loss bug

12aa565e · Gustaf Ahdritz · 49767099 · 12aa565e
Commit 12aa565e authored Oct 19, 2021 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

openfold/utils/loss.py openfold/utils/loss.py +1 -1

No files found.
--- a/openfold/utils/loss.py
+++ b/openfold/utils/loss.py
@@ -165,7 +165,7 @@ def backbone_loss(
        )

    # Take the mean over the layer dimension
-    fape_loss = torch.mean(fape_loss, dim=-1)
+    fape_loss = torch.mean(fape_loss)
    return fape_loss