Keep a logger file in sparseml.pytorch.utils

neuralmagic · Oct 17, 2023 · 049a753 · 049a753
1 parent 07dbfae
commit 049a753
Show file tree

Hide file tree

Showing 3 changed files with 1,017 additions and 971 deletions.
diff --git a/src/sparseml/core/__init__.py b/src/sparseml/core/__init__.py
@@ -20,6 +20,7 @@
 from .framework import *
 from .framework_object import *
 from .lifecycle import *
+from .logger import *
 from .model import *
 from .modifier import *
 from .optimizer import *