From db90fb1abdab2202462f41efd4c481ee029e0e19 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Fri, 28 Apr 2023 10:12:40 +0900 Subject: [PATCH] from local --- .../.data2vec_audio.py.swp | Bin 36864 -> 36864 bytes .../.model.py.swp | Bin 24576 -> 24576 bytes .../model.py | 8 ++++---- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp index 050f0ee96d8b27b70bae57ecd5a6cf0363e79e17..c770643c135ddf2b72aa3969ee4bf8f05e1b1de5 100644 GIT binary patch delta 34 ocmZozz|^pSNj%9Q%+puFQqO<^2m}}yzDawh)LO6KDE_h^0G`L+3WcD+j&5Z&zJd>aK kVI7)XJ@XMP5T6F|Hih_iv%35bP(cqbnNLk|## z0>YNY@LIl1x0 X$%#3soA)S+Fio!TZ`u6cpHTn+1NJ@L diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py index 8c219bb70..048af6161 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py @@ -84,11 +84,10 @@ class Transducer(nn.Module): print(self.sid) print('-'*20) if prompt: - self.prompt = torch.randn((200, 512), requires_grad=True) - statistic = open(f'/home/work/workspace/icefall/egs/librispeech/ASR/conv_feat/{self.sid}/{sid}_statistic.txt', 'r').readlines() + #statistic = open(f'/home/work/workspace/icefall/egs/librispeech/ASR/conv_feat/{self.sid}/{sid}_statistic.txt', 'r').readlines() self.prompt = torch.nn.Parameter(torch.rand((50, 512))) - print(self.prompt) - + #print(self.prompt) + ''' new_emb = torch.empty(512, 50) for i in range(512): mean, std = statistic[i].strip().split(' ') @@ -98,6 +97,7 @@ class Transducer(nn.Module): new_emb = new_emb.transpose(1,0) self.prompt = torch.nn.Parameter(new_emb) print(self.prompt) + ''' def forward( self,