From 6ddd383370c3b264c5c56a8fafb33e5cde20cfc5 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 08 四月 2024 21:03:49 +0800
Subject: [PATCH] bugfix (#1594)

---
 examples/industrial_data_pretraining/llm_asr/conf/whisper_vicuna_linear.yaml |    4 ++--
 examples/industrial_data_pretraining/llm_asr/conf/whisper_qwen_linear.yaml   |    4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/examples/industrial_data_pretraining/llm_asr/conf/whisper_qwen_linear.yaml b/examples/industrial_data_pretraining/llm_asr/conf/whisper_qwen_linear.yaml
index fb00b12..34be78d 100644
--- a/examples/industrial_data_pretraining/llm_asr/conf/whisper_qwen_linear.yaml
+++ b/examples/industrial_data_pretraining/llm_asr/conf/whisper_qwen_linear.yaml
@@ -67,9 +67,9 @@
    lr: 0.0001
    weight_decay: 0.000000
 
-scheduler: custom_lambdalr
+scheduler: warmuplr
 scheduler_conf:
-   warmup_steps: 1000
+   warmup_steps: 1500
 
 dataset: AudioLLMQwenAudioDataset
 dataset_conf:
diff --git a/examples/industrial_data_pretraining/llm_asr/conf/whisper_vicuna_linear.yaml b/examples/industrial_data_pretraining/llm_asr/conf/whisper_vicuna_linear.yaml
index 8a01d14..f4f8c01 100644
--- a/examples/industrial_data_pretraining/llm_asr/conf/whisper_vicuna_linear.yaml
+++ b/examples/industrial_data_pretraining/llm_asr/conf/whisper_vicuna_linear.yaml
@@ -67,9 +67,9 @@
    lr: 0.0001
    weight_decay: 0
 
-scheduler: custom_lambdalr
+scheduler: warmuplr
 scheduler_conf:
-   warmup_steps: 1000
+   warmup_steps: 1500
 
 dataset: AudioLLMVicunaDataset
 dataset_conf:

--
Gitblit v1.9.1