Skip to content

Commit 590f5f1

Browse files
committed
fix tricks Conv1dFlatWeights
Signed-off-by: Hao Wu <[email protected]>
1 parent e6ba090 commit 590f5f1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

emerging_optimizers/utils/modules.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -129,5 +129,5 @@ def extra_repr(self) -> str:
129129
s += ", bias=False"
130130
if self.padding_mode != "zeros":
131131
s += f", padding_mode={self.padding_mode}"
132-
s += f", flattened_param_shape={self.weight.shape}"
132+
s += f", flattened_param_shape={tuple(self.weight.shape)}"
133133
return s

0 commit comments

Comments
 (0)