feature(nyz): add basic mem profiler

opendilab · Apr 8, 2024 · 8a5a998 · 8a5a998
1 parent 7d05491
commit 8a5a998
Show file tree

Hide file tree

Showing 4 changed files with 763 additions and 0 deletions.
diff --git a/ding/utils/__init__.py b/ding/utils/__init__.py
@@ -32,6 +32,7 @@
 from .fast_copy import fastcopy
 from .bfs_helper import get_vi_sequence
 from .normalizer_helper import DatasetNormalizer
+from .memory_helper import SimpleMemoryProfiler
 
 if ding.enable_linklink:  # False as default
     from .linklink_dist_helper import get_rank, get_world_size, dist_mode, dist_init, dist_finalize, \