Hamid Shojanazeri преди 1 година
родител
ревизия
b15ffeeaf4
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      src/llama_recipes/finetuning.py

+ 1 - 1
src/llama_recipes/finetuning.py

@@ -45,9 +45,9 @@ from llama_recipes.utils.train_utils import (
     get_policies
 )
 
-import gc
 def main(**kwargs):
     gc.disable()
+    gc.collect(1)
     # Update the configuration for the training and sharding process
     update_config((train_config, fsdp_config), **kwargs)