mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
473dea7804
commit
025486c789
Binary file not shown.
Binary file not shown.
@ -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]):
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user