Skip to content

Commit ba6a8ec

Browse files
committed
FIX TCL
1 parent 11cd9e4 commit ba6a8ec

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tltorch/factorized_layers/tensor_contraction_layers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def reset_parameters(self):
8888
This may be renamed to init_from_random for consistency with TensorModules
8989
"""
9090
for i in range(self.order):
91-
init.kaiming_uniform_(self.factors[i], a=math.sqrt(5))
91+
init.kaiming_uniform_(getattr(self, f'factor_{i}'), a=math.sqrt(5))
9292
if self.bias is not None:
9393
bound = 1 / math.sqrt(self.input_shape[0])
9494
init.uniform_(self.bias, -bound, bound)

0 commit comments

Comments
 (0)