Hamid Shojanazeri 1 năm trước cách đây
mục cha
commit
b15ffeeaf4
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  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)