from local

This commit is contained in:
dohe0342 2023-04-27 16:36:59 +09:00
parent 473dea7804
commit 025486c789
3 changed files with 4 additions and 0 deletions

View File

@ -6,6 +6,7 @@
import argparse import argparse
import logging import logging
import math import math
import os
from dataclasses import dataclass, field from dataclasses import dataclass, field
from typing import Optional from typing import Optional
@ -492,6 +493,9 @@ class Data2VecAudioModel(BaseFairseqModel):
## for prompt tuning ## for prompt tuning
if prompt is not None: if prompt is not None:
if 1: if 1:
spk_dir = f'/home/work/workspace/icefall/egs/librispeech/ASR/conv_feat/{sid}'
if not os.path.isdir(spk_dir):
os.mkdir(spk_dir)
conv_feat_all = torch.tensor([]).to(features.device) conv_feat_all = torch.tensor([]).to(features.device)
length = 0 length = 0
for i in range(padding_mask.size()[0]): for i in range(padding_mask.size()[0]):