From f0f3ae954dd16ea9541f7c96fac2f01a8d68aec0 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 11 Apr 2023 15:58:09 +0900 Subject: [PATCH] from local --- .../.prompt_tuning.py.swp | Bin 86016 -> 86016 bytes .../prompt_tuning.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) 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 3f73b1d0dcd21cbba5a106c691389169c0837c1b..6ce4d55ba2543e9ca7e2c4a921e536a6f33c5756 100644 GIT binary patch delta 126 zcmZozz}m2YRW!*U%+puFQqO<^2m}}y4zrl1OxY;<-kldc-jgF_ delta 105 zcmZozz}m2YRW!*U%+puFQqO<^2m}}y_OO_yq;C{`@6URMmw~}xGG~CX;txIshG#&0 z8;G|9@j@W32jX}jHUnaV&9MO&c^Mg|AJk)1nH(6mWAgbtp2>#!#+$w3w|xKrSHm2n 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 6a1a8f898..77811c8db 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 @@ -1602,7 +1602,7 @@ def run_adapter(rank, world_size, args, wb=None): ) ''' - prompt = torch.nn.Parameter(torch.randn(50, 512)).to(device) + prompt = torch.randn((50, 512), requires_grad=True).to(device) print(prompt) exit() optimizer_adapter = ScaledAdam(