Python FeatureFactory.from_config Examples

Example #1

File: model.py Project: wonwizard/NVIDIA_DeepLearningExamples

 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)

Example #2

File: preprocessing.py Project: kllmia/inference_results_v0.7

 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)