def __init__()

in src/speech_reps/featurize.py [0:0]

• 9 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self, params_file, gpu=None):
        super().__init__()

        # Load the model
        from .models.bertphone import BertPhone
        import mxnet as mx
        self._model = BertPhone('bert_12_768_12', prefix='OpenSpeechModel_')
        self._ctx = mx.gpu(gpu) if gpu is not None else mx.cpu()
        self._model.collect_params().load(str(params_file),  ignore_extra=True, ctx=self._ctx)
        self._model.hybridize(static_alloc=True)
        logging.info(self._model)