trinity.manager.checkpoint_converter module
trinity.manager.checkpoint_converter module
-
class trinity.manager.checkpoint_converter.Converter(base_model_dir: str | None = None)[source]
Bases: object
-
__init__(base_model_dir: str | None = None)[source]
-
init_base_model() → bool[source]
-
init_process_group()[source]
-
init_checkpoint_converter(checkpoint_dir) → bool[source]
-
convert(checkpoint_dir: str) → None[source]
