trinity.manager.checkpoint_converter module

trinity.manager.checkpoint_converter module#

class trinity.manager.checkpoint_converter.Converter(base_model_dir: str | None = None)[源代码]#

基类：object

__init__(base_model_dir: str | None = None)[源代码]#
init_base_model() → bool[源代码]#
init_process_group()[源代码]#
init_checkpoint_converter(checkpoint_dir) → bool[源代码]#
convert(checkpoint_dir: str) → None[源代码]#
目录