@HarryVolek I have get the model of TIMIT database ,and get thi information
"Done, trained model saved at ./speech_id_checkpoint/final_epoch_950_batch_id_141.model"
But when I do test , I cant load the trained model.The log is
Traceback (most recent call last):
File "train_speech_embedder.py", line 168, in
test('speech_id_checkpoint/ckpt_epoch_840_batch_id_141.pth')
File "train_speech_embedder.py", line 109, in test
embedder_net.load_state_dict(torch.load(model_path))
File "/media/diskc/wq/.local/lib/python3.5/site-packages/torch/nn/modules/module.py", line 721, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for SpeechEmbedder:
Missing key(s) in state_dict: "LSTM_stack.weight_ih_l1", "LSTM_stack.bias_hh_l1", "LSTM_stack.bias_hh_l0", "LSTM_stack.bias_ih_l0", "LSTM_stack.weight_hh_l2", "LSTM_stack.weight_hh_l1", "LSTM_stack.bias_ih_l1", "LSTM_stack.weight_ih_l0", "LSTM_stack.weight_ih_l2", "LSTM_stack.weight_hh_l0", "LSTM_stack.bias_hh_l2", "LSTM_stack.bias_ih_l2", "projection.weight", "projection.bias".
Unexpected key(s) in state_dict: "module.LSTM_stack.weight_ih_l0", "module.LSTM_stack.weight_hh_l0", "module.LSTM_stack.bias_ih_l0", "module.LSTM_stack.bias_hh_l0", "module.LSTM_stack.weight_ih_l1", "module.LSTM_stack.weight_hh_l1", "module.LSTM_stack.bias_ih_l1", "module.LSTM_stack.bias_hh_l1", "module.LSTM_stack.weight_ih_l2", "module.LSTM_stack.weight_hh_l2", "module.LSTM_stack.bias_ih_l2", "module.LSTM_stack.bias_hh_l2", "module.projection.weight", "module.projection.bias".
@HarryVolek I have get the model of TIMIT database ,and get thi information "Done, trained model saved at ./speech_id_checkpoint/final_epoch_950_batch_id_141.model" But when I do test , I cant load the trained model.The log is Traceback (most recent call last): File "train_speech_embedder.py", line 168, in
test('speech_id_checkpoint/ckpt_epoch_840_batch_id_141.pth')
File "train_speech_embedder.py", line 109, in test
embedder_net.load_state_dict(torch.load(model_path))
File "/media/diskc/wq/.local/lib/python3.5/site-packages/torch/nn/modules/module.py", line 721, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for SpeechEmbedder:
Missing key(s) in state_dict: "LSTM_stack.weight_ih_l1", "LSTM_stack.bias_hh_l1", "LSTM_stack.bias_hh_l0", "LSTM_stack.bias_ih_l0", "LSTM_stack.weight_hh_l2", "LSTM_stack.weight_hh_l1", "LSTM_stack.bias_ih_l1", "LSTM_stack.weight_ih_l0", "LSTM_stack.weight_ih_l2", "LSTM_stack.weight_hh_l0", "LSTM_stack.bias_hh_l2", "LSTM_stack.bias_ih_l2", "projection.weight", "projection.bias".
Unexpected key(s) in state_dict: "module.LSTM_stack.weight_ih_l0", "module.LSTM_stack.weight_hh_l0", "module.LSTM_stack.bias_ih_l0", "module.LSTM_stack.bias_hh_l0", "module.LSTM_stack.weight_ih_l1", "module.LSTM_stack.weight_hh_l1", "module.LSTM_stack.bias_ih_l1", "module.LSTM_stack.bias_hh_l1", "module.LSTM_stack.weight_ih_l2", "module.LSTM_stack.weight_hh_l2", "module.LSTM_stack.bias_ih_l2", "module.LSTM_stack.bias_hh_l2", "module.projection.weight", "module.projection.bias".