ColossalAI/colossalai/utils/memory_tracer/__init__.py

4 lines
82 B
Python

from .async_memtracer import AsyncMemoryMonitor
__all__ = ['AsyncMemoryMonitor']