12 lines
370 B
Python
12 lines
370 B
Python
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,hidden_dim,wav_frame]
|
|
'''
|
|
pass |