Merge pull request #23 from eltociear/patch-1

synxlin · web-flow · commit c53abf3fe343 · 2025-02-14T00:22:02.000-05:00
chore: update search.py
diff --git a/deepcompressor/calib/search.py b/deepcompressor/calib/search.py
@@ -647,7 +647,7 @@ def calibrate(
             eval_kwargs=eval_kwargs,
             **kwargs,
         )
-        self.logger.debug(f"+ finished reseting calibrator, ram usage: {psutil.virtual_memory().percent}")
+        self.logger.debug(f"+ finished resetting calibrator, ram usage: {psutil.virtual_memory().percent}")
         gc.collect()
         torch.cuda.empty_cache()
         if self.tensor_type == TensorType.Weights:

Original file line number	Diff line number	Diff line change
`@@ -647,7 +647,7 @@ def calibrate(`
`647`	`647`	`eval_kwargs=eval_kwargs,`
`648`	`648`	`**kwargs,`
`649`	`649`	`)`
`650`		`- self.logger.debug(f"+ finished reseting calibrator, ram usage: {psutil.virtual_memory().percent}")`
	`650`	`+ self.logger.debug(f"+ finished resetting calibrator, ram usage: {psutil.virtual_memory().percent}")`
`651`	`651`	`gc.collect()`
`652`	`652`	`torch.cuda.empty_cache()`
`653`	`653`	`if self.tensor_type == TensorType.Weights:`