add LoRACheckpointToSaveHook

open-mmlab · Sep 23, 2023 · 0e6356a · 0e6356a
1 parent bb2cbec
commit 0e6356a
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/tests/test_engine/test_hooks/test_lora_checkpoint_to_save_hook.py b/tests/test_engine/test_hooks/test_lora_checkpoint_to_save_hook.py
@@ -1,6 +1,5 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 import copy
-import gc
 import platform
 
 import pytest
@@ -87,11 +86,6 @@ def tearDown(self):
         MODELS.module_dict.pop('UNet2DConditionModel')
         MODELS.module_dict.pop('AutoencoderKL')
         MODELS.module_dict.pop('DataPreprocessor')
-
-        gc.collect()
-        globals().clear()
-        locals().clear()
-
         return super().tearDown()
 
     def test_init(self):