File size: 370 Bytes
f2cd0c4
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
class SpeechEncoder(object):
    def __init__(self,vec_path = "pretrain/checkpoint_best_legacy_500.pt",device=None):
        self.model = None  #This is Model
        self.hidden_dim = 768
        pass

    def encoder(self,wav):
        '''
        input: wav:[batchsize,signal_length]
        output: embedding:[batchsize,wav_frame,hidden_dim]
        '''
        pass