Fix OPM bug

ad2e5c97 · Gustaf Ahdritz · b2d102cb · ad2e5c97
Commit ad2e5c97 authored Sep 27, 2021 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

No files found.
--- a/openfold/model/outer_product_mean.py
+++ b/openfold/model/outer_product_mean.py
@@ -107,7 +107,7 @@ class OuterProductMean(nn.Module):
        norm = torch.einsum("...abc,...adc->...bdc", mask, mask)

        # [*, N_res, N_res, C_z]
-        outer = outer / self.eps + norm
+        outer = outer / (self.eps + norm)

        return outer