This commit is contained in:
@@ -0,0 +1,16 @@
|
||||
# Copyright (c) ModelScope Contributors. All rights reserved.
|
||||
from .base import CheckpointEngine, MasterMetadata, TensorMeta
|
||||
from .hccl import HCCLCheckpointEngine
|
||||
from .manager import CheckpointEngineManager
|
||||
from .mixin import CheckpointEngineMixin
|
||||
from .nccl import NCCLCheckpointEngine
|
||||
|
||||
__all__ = [
|
||||
'CheckpointEngine',
|
||||
'MasterMetadata',
|
||||
'TensorMeta',
|
||||
'CheckpointEngineMixin',
|
||||
'CheckpointEngineManager',
|
||||
'NCCLCheckpointEngine',
|
||||
'HCCLCheckpointEngine',
|
||||
]
|
||||
Reference in New Issue
Block a user