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