From 9b3c4f8d50020ded33621d7d8d05a59c7f8eb110 Mon Sep 17 00:00:00 2001
From: Eda Z <36808054+edamamez@users.noreply.github.com>
Date: Tue, 3 Sep 2024 15:38:27 -0700
Subject: [PATCH] Update get_default_finetune_args.py
---
.../text2sql_memory_tuning/util/get_default_finetune_args.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
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 6cddb41a..3d24dae5 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},
--
GitLab