コード例 #1
0
ファイル: torch_component.py プロジェクト: zhoumo99133/HanLP
    def load_config(self, save_dir, filename='config.json', **kwargs):
        """Load config from a directory.

        Args:
            save_dir: The directory to load config.
            filename: A file name for config.
            **kwargs: K-V pairs to override config.
        """
        save_dir = get_resource(save_dir)
        self.config.load_json(os.path.join(save_dir, filename))
        self.config.update(kwargs)  # overwrite config loaded from disk
        for k, v in self.config.items():
            if isinstance(v, dict) and 'classpath' in v:
                self.config[k] = Configurable.from_config(v)
        self.on_config_ready(**self.config)
コード例 #2
0
 def _convert_embed(self):
     embed = self.config['embed']
     if isinstance(embed, dict):
         self.config['embed'] = embed = Configurable.from_config(embed)
     return embed