Gastron
/

foreign-class-example

Model card Files Files and versions Community

Aku Rouhe commited on Nov 20, 2021

Commit

85046ed

•

1 Parent(s): fa827db

New naming scheme

Files changed (1) hide show

interface.py → custom.py +1 -1

interface.py → custom.py RENAMED Viewed

@@ -9,7 +9,7 @@ class FeatureScaler(torch.nn.Module):
     def forward(x):
         return x * self.scaler
-class Custom(sb.pretrained.interfaces.Pretrained):
     MODULES_NEEDED = ["normalizer"]
     HPARAMS_NEEDED = ["feature_extractor"]

     def forward(x):
         return x * self.scaler
+class CustomInterface(sb.pretrained.interfaces.Pretrained):
     MODULES_NEEDED = ["normalizer"]
     HPARAMS_NEEDED = ["feature_extractor"]