Add gc.collect before compress call

openvinotoolkit · Jul 26, 2024 · d11b3b7 · d11b3b7
1 parent b5e4267
commit d11b3b7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/tests/post_training/pipelines/base.py b/tests/post_training/pipelines/base.py
@@ -9,6 +9,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import datetime as dt
+import gc
 import os
 import re
 import time
@@ -357,6 +358,7 @@ def compress(self) -> None:
 
         start_time = time.perf_counter()
         if self.memory_monitor:
+            gc.collect()
             max_value_per_memory_type = monitor_memory_for_callable(
                 self._compress,
                 interval=0.1,

diff --git a/tests/post_training/pipelines/lm_weight_compression.py b/tests/post_training/pipelines/lm_weight_compression.py
@@ -8,7 +8,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
+import gc
 import os
 import re
 import shutil
@@ -182,6 +182,7 @@ def compress(self) -> None:
         print("Weight compression...")
         start_time = time.perf_counter()
         if self.memory_monitor:
+            gc.collect()
             max_value_per_memory_type = monitor_memory_for_callable(
                 self._compress,
                 interval=0.1,