Explorar el Código

Update get_default_finetune_args.py

Eda Z hace 7 meses
padre
commit
9b3c4f8d50

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