Skip to content

Commit 29f9897

Browse files
committed
equalizers: Fix trainable argument evaluation in Butterfly2x2
1 parent 7a507b9 commit 29f9897

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/mokka/equalizers/torch.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ def __init__(self, taps=None, num_taps=None, trainable=False, timedomain=True):
114114
# We store h_xx, h_xy, h_yy, h_yx
115115
if trainable:
116116
self.register_parameter("taps", torch.nn.Parameter(filter_taps))
117-
elif trainable:
117+
elif not trainable:
118118
self.register_buffer("taps", filter_taps)
119119
else:
120120
raise ValueError("trainable must be set to either True or False.")

0 commit comments

Comments
 (0)