소스 검색

Update get_default_finetune_args.py

Eda Z 1 년 전
부모
커밋
9b3c4f8d50
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      recipes/3p_integrations/lamini/text2sql_memory_tuning/util/get_default_finetune_args.py

+ 2 - 2
recipes/3p_integrations/lamini/text2sql_memory_tuning/util/get_default_finetune_args.py

@@ -1,7 +1,7 @@
 def get_default_finetune_args():
     return {
-        "learning_rate": 3e-4,
-        "max_steps": 360,
+        "learning_rate": 0.0003,
+        "max_steps": 60,
         "early_stopping": False,
         "load_best_model_at_end": False,
         "peft_args": {"r_value": 32},