Skip to content

Commit

Permalink
fix the lint error
Browse files Browse the repository at this point in the history
  • Loading branch information
wsstriving authored and wsstriving committed Aug 27, 2024
1 parent c347bd0 commit 7100c4c
Showing 1 changed file with 5 additions and 7 deletions.
12 changes: 5 additions & 7 deletions wespeaker/models/redimnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -141,9 +141,7 @@ def __init__(
self,
C,
dim=2,
kernel_sizes=[
(3, 3),
],
kernel_sizes=((3, 3),),
group_divisor=1,
padding="same",
):
Expand Down Expand Up @@ -612,7 +610,7 @@ def __init__(
C=16,
block_1d_type="conv+att",
block_2d_type="basic_resnet",
stages_setup=[
stages_setup=(
# stride, num_blocks, conv_exp, kernel_size, att_block_red
(1, 2, 1, [(3, 3)], None), # 16
(2, 3, 1, [(3, 3)], None), # 32
Expand All @@ -621,7 +619,7 @@ def __init__(
(2, 5, 1, [(3, 3)], 8), # 128
(1, 5, 1, [(7, 1)], 8), # 128 # TDNN - time context
(2, 3, 1, [(3, 3)], 8), # 256
],
),
group_divisor=1,
out_channels=512,
):
Expand Down Expand Up @@ -785,15 +783,15 @@ def __init__(
block_1d_type="conv+att",
block_2d_type="basic_resnet",
# Default setup: M version:
stages_setup=[
stages_setup=(
# stride, num_blocks, kernel_sizes, layer_ext, att_block_red
(1, 2, 1, [(3, 3)], 12),
(2, 2, 1, [(3, 3)], 12),
(1, 3, 1, [(3, 3)], 12),
(2, 4, 1, [(3, 3)], 8),
(1, 4, 1, [(3, 3)], 8),
(2, 4, 1, [(3, 3)], 4),
],
),
group_divisor=4,
out_channels=None,
# -------------------------
Expand Down

0 comments on commit 7100c4c

Please sign in to comment.