From 16c41542451f399bdb716f1d7cad31cf52f6f8c3 Mon Sep 17 00:00:00 2001
From: haoneng.lhn <haoneng.lhn@alibaba-inc.com>
Date: 星期二, 18 七月 2023 16:47:27 +0800
Subject: [PATCH] add lora finetune code

---
 funasr/bin/build_trainer.py |    2 ++
 1 files changed, 2 insertions(+), 0 deletions(-)

diff --git a/funasr/bin/build_trainer.py b/funasr/bin/build_trainer.py
index 891139a..bd30a83 100644
--- a/funasr/bin/build_trainer.py
+++ b/funasr/bin/build_trainer.py
@@ -144,6 +144,8 @@
         args.patience = None
     args.local_rank = local_rank
     args.distributed = distributed
+    for key, value in kwargs.items():
+        args.key = value
     ASRTask.finetune_args = args
 
     return ASRTask

--
Gitblit v1.9.1