import logger from sparseml.core

neuralmagic · Oct 15, 2023 · 5eb2b77 · 5eb2b77
1 parent 3558e03
commit 5eb2b77
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/neuralmagic/sparsification_manager.py b/utils/neuralmagic/sparsification_manager.py
@@ -4,7 +4,7 @@
 
 import torch
 from sparseml.pytorch.optim import ScheduledModifierManager
-from sparseml.pytorch.utils import SparsificationGroupLogger
+from sparseml.core.logger import SparsificationGroupLogger
 
 from utils.loggers import Loggers
 from utils.loss import ComputeLoss