From 5825bbfe33e697f67a7e036224381171a311d44e Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 11 Apr 2023 15:35:08 +0900 Subject: [PATCH] from local --- .../.prompt_tuning.py.swp | Bin 81920 -> 81920 bytes .../prompt_tuning.py | 2 ++ 2 files changed, 2 insertions(+) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp index fcfd1bdf4247ffbbf99b4199244727bb2a31c997..dd49f49404b4db964d81b6b8fd724c34a4348071 100644 GIT binary patch delta 94 zcmZo@U~On%6-zP*^Ym4))H7fJ0s#hwDgRAUoTqFQ+a<`zz4@Y`u^;auJ_d$YKn#-Q so-CN4&H5E2ve_|VE#Ks;nS%TUMfte}C3?k~RjC@9nhcxy@;_Yw0I{kZ1ONa4 delta 85 zcmZo@U~On%6-zP*^Ym4))H7fJ0s#hwy1ynVZ~Hfj?Gj|<+nEVd+s%#%Yx&p|6ch>+6ejcKOKs-M|8xNWvey?V diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py index 8ec4a5462..ac46a1fb2 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py @@ -1592,6 +1592,8 @@ def run_adapter(rank, world_size, args, wb=None): p.requires_grad = False prompt = torch.nn.Parameter(torch.randn(50, 512)).to(device) + print(prompt.size()) + exit() ''' optimizer_adapter = ScaledAdam(