From b303ca3d7d88ea3c02d469a0bdcf978fc2b4be20 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Wed, 24 May 2023 13:51:29 +0900 Subject: [PATCH] from local --- .../.train_lora.py.swp | Bin 94208 -> 94208 bytes .../train_lora.py | 5 ++++- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp index c2aad1af17de0b8d3c4cb215e805a596a88ca5db..3f33d0488d305409b8c74b9374319d934e631278 100644 GIT binary patch delta 165 zcmZp8z}oPDRW!*U%+puFQqO<^2m}}ymQBx1>DwqOEyO6^Y$&wdP>AueF7F8*1_lRS zhywBHf)0$@W*7Mw7}f#t0wAsd;%XpH1maL2_61@kAQlAT!$58Gfw&ro9e6h{6k5VF peVs3()^;G_!Z`h@J)_w4llF|}TnbQ-nKnJZkI{HLpFiU#1psuuBr*U1 delta 157 zcmZp8z}oPDRW!*U%+puFQqO<^2m}}yvZm*zL~Rt67Ge}>HWb=!D8zVKm)D7xfngmG zgA|BN7j$6M)_lds!0-}?&jaycAl?nc6M(oLhz)_57l>biv~4eRU|h;G@xhYG`zCTs tm-A&*n&!tSIsIfjqu_Kue@64^r+pcBPJe07C^UVRA0yXxK7Ynf3IJaxD24z4 diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py index 01dd11ec6..9fd2618ec 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py @@ -1605,7 +1605,10 @@ def run_adapter(rank, world_size, args, wb=None): adapter_param.append(p) for n, p in model.named_parameters(): - p.requires_grad = False + if 'joiner' in n or 'simple' in n or 'ctc' in n: + p.requires_grad = True + else: + p.requires_grad = False #print(adapter_names) #exit()