Skip to content

Commit

Permalink
FIX TCL
Browse files Browse the repository at this point in the history
  • Loading branch information
JeanKossaifi committed Jan 10, 2023
1 parent 11cd9e4 commit ba6a8ec
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tltorch/factorized_layers/tensor_contraction_layers.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ def reset_parameters(self):
This may be renamed to init_from_random for consistency with TensorModules
"""
for i in range(self.order):
init.kaiming_uniform_(self.factors[i], a=math.sqrt(5))
init.kaiming_uniform_(getattr(self, f'factor_{i}'), a=math.sqrt(5))
if self.bias is not None:
bound = 1 / math.sqrt(self.input_shape[0])
init.uniform_(self.bias, -bound, bound)

0 comments on commit ba6a8ec

Please sign in to comment.