[Logger Refactor] Add timer in logger manager (#1967)

* Add timer in logger manager

* Address review comments

* update docstring

* fix argument

src/sparseml/core/logger/logger.py

@@ -21,6 +21,7 @@
 import time
 import warnings
 from abc import ABC
+from contextlib import contextmanager
 from datetime import datetime
 from logging import CRITICAL, DEBUG, ERROR, INFO, WARN, Logger
 from pathlib import Path
@@ -1074,6 +1075,25 @@ def save(
             if log.enabled:
                 log.save(file_path, **kwargs)
 
+    @contextmanager
+    def time(self, tag: Optional[str] = None, *args, **kwargs):
+        """
+        Context manager to log the time it takes to run the block of code
+
+        Usage:
+        >>> with LoggerManager().time("my_block"):
+        >>>    time.sleep(1)
+
+        :param tag: identifying tag to log the values with
+        """
+
+        start = time.time()
+        yield
+        elapsed = time.time() - start
+        if not tag:
+            tag = f"{DEFAULT_TAG}_time_secs"
+        self.log_scalar(tag=tag, value=float(f"{elapsed:.3f}"), *args, **kwargs)
+
 
 class LoggingWrapperBase:
     """