Fix another initialization bug

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

Showing with 2 additions and 3 deletions

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

No files found.
--- a/openfold/model/primitives.py
+++ b/openfold/model/primitives.py
@@ -34,9 +34,8 @@ def _prod(nums):
    return out
-def _calculate_fan(shape, fan="fan_in"):
+def _calculate_fan(linear_weight_shape, fan="fan_in"):
-    fan_in = shape[-2]
+    fan_out, fan_in = linear_weight_shape
-    fan_out = shape[-1]
    if fan == "fan_in":
        f = fan_in