Gastron
/

foreign-class-example

Model card Files Files and versions

Aku Rouhe commited on Nov 20, 2021

Commit

0f77019

·

1 Parent(s): 8e970cf

small fix

Files changed (1) hide show

custom.py +1 -1

custom.py CHANGED Viewed

@@ -6,7 +6,7 @@ class FeatureScaler(torch.nn.Module):
         super().__init__()
         self.scaler = torch.eye(num_in) * scale
-    def forward(x):
         return x * self.scaler
 class CustomInterface(sb.pretrained.interfaces.Pretrained):

         super().__init__()
         self.scaler = torch.eye(num_in) * scale
+    def forward(self, x):
         return x * self.scaler
 class CustomInterface(sb.pretrained.interfaces.Pretrained):