def __init__(self, **kwargs): nn.Module.__init__(self) # For PyTorch API self.optim_level = kwargs.get('optimization_level', 0) self.featurizer = FeatureFactory.from_config(kwargs) self.transpose_out = kwargs.get("transpose_out", False)
def __init__(self, **kwargs): nn.Module.__init__(self) # For PyTorch API self.optim_level = kwargs.get('optimization_level', Optimization.nothing) self.featurizer = FeatureFactory.from_config(kwargs)