def load_model()

in src/neo_loader/mxnet_model_loader.py [0:0]

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

    def load_model(self) -> None:
        logger.info("Generating relay IR for mxnet model!")
        model_json = self.__get_model_json_from_model_artifact()
        arg_params, aux_params = self.__get_arg_and_aux_params_from_model_artifact()
        compiler_option = json.loads(os.environ.get('COMPILER_OPTIONS')) if os.environ.get('COMPILER_OPTIONS') is not None else None
        current_data_shape = self.data_shape['data'] if "data" in self.data_shape else []
        if current_data_shape[-1:] == [-1] and compiler_option:
            if compiler_option.get("PLATFORM") == "AL2012":
                feature_dim = self.__check_feature_dim()
                if feature_dim:
                    keys = list(self.data_shape.keys())
                    current_data_shape[-1] = feature_dim
                    self.data_shape[keys[-1]] = current_data_shape
        try:
            self._relay_module_object, self._params = relay.frontend.from_mxnet(
                model_json,
                self.data_shape,
                arg_params=arg_params,
                aux_params=aux_params
            )
            self.update_missing_metadata()
        except OpError:
            raise
        except Exception as e:
            logger.exception("Failed to convert mxnet model. %s" % repr(e))
            raise RuntimeError("InputConfiguration: TVM can't convert the MXNet model. {}".format(e))
        else:
            logger.info("Successfully generated relay IR for mxnet model!")