ColossalAI/colossalai/zero/init_ctx/__init__.py

mirror of https://github.com/hpcaitech/ColossalAI
from .init_context import ZeroInitContext, no_shard_zero_context, no_shard_zero_decrator

__all__ = ['ZeroInitContext', 'no_shard_zero_context', 'no_shard_zero_decrator']
	`from .init_context import ZeroInitContext, no_shard_zero_context, no_shard_zero_decrator`

	`__all__ = ['ZeroInitContext', 'no_shard_zero_context', 'no_shard_zero_decrator']`