Skip to content

Commit

Permalink
Revert "Temporary enable memory monitor by default"
Browse files Browse the repository at this point in the history
This reverts commit d53d9cb.
  • Loading branch information
nikita-savelyevv committed Jul 17, 2024
1 parent d53d9cb commit 96c75a5
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion tests/post_training/pipelines/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -356,7 +356,7 @@ def compress(self) -> None:
torch.set_num_threads(int(inference_num_threads))

start_time = time.perf_counter()
if self.memory_monitor or True:
if self.memory_monitor:
max_value_per_memory_type = monitor_memory_for_callable(
self._compress,
interval=0.1,
Expand Down
2 changes: 1 addition & 1 deletion tests/post_training/pipelines/lm_weight_compression.py
Original file line number Diff line number Diff line change
Expand Up @@ -179,7 +179,7 @@ def compress(self) -> None:

print("Weight compression...")
start_time = time.perf_counter()
if self.memory_monitor or True:
if self.memory_monitor:
max_value_per_memory_type = monitor_memory_for_callable(
self._compress,
interval=0.1,
Expand Down

0 comments on commit 96c75a5

Please sign in to comment.