from local

This commit is contained in:
dohe0342 2023-04-27 16:31:05 +09:00
parent d6db3b10fa
commit eff27197c0
2 changed files with 1 additions and 1 deletions

View File

@ -504,7 +504,7 @@ class Data2VecAudioModel(BaseFairseqModel):
conv_feat_all = torch.cat([conv_feat_all, features[i]])
randint = np.random.randint(10000)
np.save(f'/home/work/workspace/icefall/egs/librispeech/ASR/conv_feat/{randint}.npy', conv_feat_all.cpu().numpy())
np.save(f'/home/work/workspace/icefall/egs/librispeech/ASR/conv_feat/{sid}/{randint}.npy', conv_feat_all.cpu().numpy())
prompt = prompt.expand((features.size()[0], prompt.size()[0], prompt.size()[1]))
features = torch.cat([prompt, features], dim=1)