Skip to content

Commit 839ec54

Browse files
Merge pull request #4 from MSD-IRIMAS/litemv-fix-conv
[BUG] LITEMV replace Conv1D with DepthwiseConv1D
2 parents dd9ef94 + cb2eff1 commit 839ec54

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

classifiers/litemv.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -86,8 +86,7 @@ def hybird_layer(
8686

8787
filter_[indices_ % 2 > 0] *= -1
8888

89-
conv = tf.keras.layers.Conv1D(
90-
filters=1,
89+
conv = tf.keras.layers.DepthwiseConv1D(
9190
kernel_size=kernel_size,
9291
padding="same",
9392
use_bias=False,
@@ -121,8 +120,7 @@ def hybird_layer(
121120
filter_[kernel_size : 5 * kernel_size // 4] = -filter_left
122121
filter_[5 * kernel_size // 4 :] = -filter_right
123122

124-
conv = tf.keras.layers.Conv1D(
125-
filters=1,
123+
conv = tf.keras.layers.DepthwiseConv1D(
126124
kernel_size=kernel_size + kernel_size // 2,
127125
padding="same",
128126
use_bias=False,

0 commit comments

Comments
 (0)