From b9ef1eb801e5e9e5c46b614d9e74176d79cff4e3 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 10 Jan 2023 01:13:42 +0900 Subject: [PATCH] from local --- .../ASR/incremental_transf/.train.py.swp | Bin 61440 -> 61440 bytes .../ASR/incremental_transf/train.py | 4 ++++ 2 files changed, 4 insertions(+) diff --git a/egs/librispeech/ASR/incremental_transf/.train.py.swp b/egs/librispeech/ASR/incremental_transf/.train.py.swp index 6a7d650f8e000ce0d929ca156fccde9f02cc3f20..5dc3a5e579dd61128d2033b35d045c9cfb9ad9ad 100644 GIT binary patch delta 264 zcmZp8z})bFSuDvQ%+puFQqO<^2m}}yKG^I@R(9SfCMC!iv)NGaav$#vMh1oyW{7Oe zX2AvQjP-Zf85oWO@g^YN2*m9`oCU;IK>V7Gf#EI?uLt64Aa(=d->eJ_hk>{Qi1mP2 z28f@qFfdF9;&>od0pe3YD~&Z?q-%1f7}7r2t8K- delta 221 zcmZp8z})bFSuDvQ%+puFQqO<^2m}}y?%3=}e(AJPOiGY3VzZ&(a448*2D%nrnt*%%n+0dX)8O9Sx(RtAPPAT9x78z6qq!oaW{ zh--jY35aEZm;;DE1C74{#Cw4lWYsYsb_1%n2VyQDJ_l5D7KqmXaRm#*=IIO1GENqM c6~v|h29xU-OKq-OY{4?wX^GHgg%uCn0X7va`v3p{ diff --git a/egs/librispeech/ASR/incremental_transf/train.py b/egs/librispeech/ASR/incremental_transf/train.py index f740e1167..19e81e32b 100755 --- a/egs/librispeech/ASR/incremental_transf/train.py +++ b/egs/librispeech/ASR/incremental_transf/train.py @@ -974,6 +974,10 @@ def run(rank, world_size, args): pre_trained2 = torch.load(path2) pre_trained_model2 = pre_trained2["model"] + for n, p in pre_trained_model2.items(): + if rank == 0: + print(n) + for n, p in model.named_parameters(): if 'layer' not in n: try: p.data = pre_trained_model1[n]