diff --git a/recipes/3p_integrations/lamini/text2sql_memory_tuning/util/get_default_finetune_args.py b/recipes/3p_integrations/lamini/text2sql_memory_tuning/util/get_default_finetune_args.py index 6cddb41a6576067e59693690b34b5bc95f001de8..3d24dae56469e086554e41b794fbc66e809f6804 100644 --- a/recipes/3p_integrations/lamini/text2sql_memory_tuning/util/get_default_finetune_args.py +++ b/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},