From 056a9bed91009f8557e6f84f90bca961f206e787 Mon Sep 17 00:00:00 2001 From: Benjamin Fineran Date: Mon, 30 Aug 2021 16:59:09 -0400 Subject: [PATCH] fix qat_start bug in SparseML wrapper --- utils/sparse.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/sparse.py b/utils/sparse.py index 80dc222e7dd0..e0f53c186443 100644 --- a/utils/sparse.py +++ b/utils/sparse.py @@ -119,7 +119,7 @@ def qat_active(self, epoch): if not self.enabled or not self.manager.quantization_modifiers: return False - qat_start = max([mod.start_epoch for mod in self.manager.quantization_modifiers]) + qat_start = min([mod.start_epoch for mod in self.manager.quantization_modifiers]) return qat_start < epoch + 1 @@ -136,4 +136,4 @@ def reset_best(self, epoch): qat_start = math.floor(max([mod.start_epoch for mod in self.manager.quantization_modifiers])) \ if self.manager.quantization_modifiers else -1 - return (pruning_start <= epoch <= pruning_end) or epoch == qat_start \ No newline at end of file + return (pruning_start <= epoch <= pruning_end) or epoch == qat_start