Skip to content

Commit 5a57407

Browse files
committed
Fix weights in Independent Filter
1 parent 40f71c1 commit 5a57407

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

complexnn/init.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,9 +68,9 @@ def __call__(self, shape, dtype=None):
6868
multip_constant = np.sqrt(desired_var / np.var(independent_filters))
6969
scaled_indep = multip_constant * independent_filters
7070

71+
7172
if self.weight_dim == 2 and self.nb_filters is None:
72-
weight_real = scaled_real
73-
weight_imag = scaled_imag
73+
weight = scaled_indep
7474
else:
7575
kernel_shape = tuple(self.kernel_size) + (self.input_dim, self.nb_filters)
7676
if self.weight_dim == 1:

0 commit comments

Comments
 (0)