12 lines
358 B
Python
12 lines
358 B
Python
|
class SpeechEncoder(object):
|
||
|
def __init__(self,vec_path = "pretrain/checkpoint_best_legacy_500.pt"):
|
||
|
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
|