Преглед на файлове

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},