Merge pull request speechbrain#953 from alumae/epaca_classifier_ff_fix

mravanelli · web-flow · commit ae49958c93b7 · 2021-08-25T21:25:45.000-04:00
Small bug fix in ECAPA-TDNN classifier
diff --git a/speechbrain/lobes/models/ECAPA_TDNN.py b/speechbrain/lobes/models/ECAPA_TDNN.py
@@ -523,7 +523,7 @@ def __init__(
         for block_index in range(lin_blocks):
             self.blocks.extend(
                 [
-                    _BatchNorm1d(input_size),
+                    _BatchNorm1d(input_size=input_size),
                     Linear(input_size=input_size, n_neurons=lin_neurons),
                 ]
             )

Original file line number	Diff line number	Diff line change
`@@ -523,7 +523,7 @@ def __init__(`
`523`	`523`	`for block_index in range(lin_blocks):`
`524`	`524`	`self.blocks.extend(`
`525`	`525`	`[`
`526`		`- _BatchNorm1d(input_size),`
	`526`	`+ _BatchNorm1d(input_size=input_size),`
`527`	`527`	`Linear(input_size=input_size, n_neurons=lin_neurons),`
`528`	`528`	`]`
`529`	`529`	`)`