Rename _ResidualModuleMI

braindecode · Nov 9, 2022 · d1c03d6 · d1c03d6
1 parent d9ba027
commit d1c03d6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/braindecode/models/eeginception_mi.py b/braindecode/models/eeginception_mi.py
@@ -107,7 +107,7 @@ def __init__(
             ) for _ in range(2)
         ])
 
-        self.residual_block_1 = _ResidualBlockMI(
+        self.residual_block_1 = _ResidualModuleMI(
             in_channels=self.in_channels,
             n_filters=intermediate_in_channels,
             activation=self.activation,
@@ -124,7 +124,7 @@ def __init__(
             ) for _ in range(3)
         ])
 
-        self.residual_block_2 = _ResidualBlockMI(
+        self.residual_block_2 = _ResidualModuleMI(
             in_channels=intermediate_in_channels,
             n_filters=intermediate_in_channels,
             activation=self.activation,
@@ -256,7 +256,7 @@ def forward(
         return self.activation(out)
 
 
-class _ResidualBlockMI(nn.Module):
+class _ResidualModuleMI(nn.Module):
     def __init__(
         self,
         in_channels,