Gastron
/

foreign-class-example

Model card Files Files and versions Community

Aku Rouhe commited on Nov 20, 2021

Commit

b57ac94

•

1 Parent(s): 0f77019

Okay now

Files changed (1) hide show

custom.py +1 -1

custom.py CHANGED Viewed

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

 class FeatureScaler(torch.nn.Module):
     def __init__(self, num_in, scale):
         super().__init__()
+        self.scaler = torch.ones((num_in,))* scale
     def forward(self, x):
         return x * self.scaler