定义一个假设的 ASR 模型结构

class ASRModel(torch.nn.Module):
def init(self):
super(ASRModel, self).init()
self.lstm = torch.nn.LSTM(input_size=160, hidden_size=256, num_layers=3, batch_first=True)
self.linear = torch.nn.Linear(256, 29) # 假设有 29 个输出类用于字符

def forward(self, x):
    x, _ = self.lstm(x)
    x = self.linear(x)
    return x