Fix implementation

custom.py

@@ -4,7 +4,7 @@ import speechbrain as sb
 class FeatureScaler(torch.nn.Module):
     def __init__(self, num_in, scale):
         super().__init__()
-        self.scaler = torch.nn.eye(num_in) * scale
+        self.scaler = torch.eye(num_in) * scale
 
     def forward(x):
         return x * self.scaler