Fix DRMSD bug

4dd50722 · Gustaf Ahdritz · 0a0a741f · 4dd50722
Commit 4dd50722 authored Oct 08, 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
@@ -1340,7 +1340,7 @@ def compute_drmsd(structure_1, structure_2):
    drmsd = d1 - d2
    drmsd = drmsd ** 2
    drmsd = torch.sum(drmsd, dim=(-1, -2))
-    n = structure_1.shape[-1]
+    n = d1.shape[-1]
    drmsd = drmsd * (1 / (n * (n - 1)))
    drmsd = torch.sqrt(drmsd)