def __get_arg_and_aux_params_from_model_artifact()

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

• 14 lines of code
• 6 McCabe index (conditional complexity)

    def __get_arg_and_aux_params_from_model_artifact(self) -> (Dict, Dict):
        saved_dict = self.__get_saved_dict_from_model_artifact()
        arg_params, aux_params = {}, {}

        for key, value in saved_dict.items():
            if not key.startswith('arg:') and not key.startswith('aux:'):
                raise RuntimeError("InputConfiguration: Framework can't load the MXNet model: Please use HybridBlock.export() "
                                   "or gluoncv.utils.export_block() to export your model instead of "
                                   "Block.save_parameters().")

            prefix, name = key.split(':', 1)
            if prefix == "arg":
                arg_params[name] = value
            elif prefix == "aux":
                aux_params[name] = value

        return arg_params, aux_params