Thanks, the code works for me:
model = AudioLSTM(1, 1, 0, 1, 1)
sd = model.state_dict()
torch.save(sd, 'tmp.pt')

model = AudioLSTM(1, 1, 0, 1, 1)
model.load_state_dict(torch.load('tmp.pt'))
> <All keys matched successfully>