From 00ca422eb889d6d721249243f64b69cb06dc035e Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Wed, 24 May 2023 13:40:04 +0900 Subject: [PATCH] from local --- .../.train_lora.py.swp | Bin 90112 -> 90112 bytes .../train_lora.py | 5 +++++ 2 files changed, 5 insertions(+) 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 0501bf10ff76a6863366be6e94f8b3b675166bef..1a14915313216fcd472c05dd1c2547cfbb73bbb3 100644 GIT binary patch delta 208 zcmZoTz}j$tRW!*U%+puFQqO<^2m}}y7^mi@cx@E@EyyU+EGV>HP>AufF0ThK1H%>| z1}PAkF6h9hY-GvLz`zg0@Aw!P-U9IzAU+Ai2Z4AR5Vr!cIS>m2@q4iL&5c5Syp#7$ zC-H9gOd(R_QCKVz8y08raBb^rhX delta 122 zcmZoTz}j$tRW!*U%+puFQqO<^2m}}yPEE;85!fjDTab~zSx{)Zpb+C{UEV{y3=C3y z5C#0x1sxcb<@EU(7}SCI4^a3u5YGkTN+1peVky4OjY9Rj6CZd?-ZznR`yO9L8OG`2 OevAU!{rnj>2><}Ewj7`U 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 b9d62817f..f5f70344f 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 @@ -1603,6 +1603,11 @@ def run_adapter(rank, world_size, args, wb=None): for n, p in lora.named_parameters(): adapter_names.append(n) adapter_param.append(p) + + for n, p in model.named_parameters(): + p..requires_grad = False + + print(adapter_names) exit() ''' for n, p in model.named_parameters():