yl12053's picture
FC
bfd2c22
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