Fix initialization bug

89abce8f · Gustaf · 8c95493f · 89abce8f
Commit 89abce8f authored Nov 10, 2021 by Gustaf
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 5 deletions

openfold/model/primitives.py openfold/model/primitives.py +2 -5

No files found.
--- a/openfold/model/primitives.py
+++ b/openfold/model/primitives.py
@@ -35,11 +35,8 @@ def _prod(nums):


 def _calculate_fan(shape, fan="fan_in"):
-    i = shape[0]
-    o = shape[1]
-    prod = _prod(shape[:2])
-    fan_in = prod * i
-    fan_out = prod * o
+    fan_in = shape[-2]
+    fan_out = shape[-1]

    if fan == "fan_in":
        f = fan_in