Closed TheisTrue closed 4 years ago
When I ran the code, I had a different task in my hand in which I did not need a really good model. Hence I did not train the model to good results. You can find the pre-trained model at - https://drive.google.com/open?id=1KOpQLAnYfiVb2FdRPo002ToaatOtV8Ww.
Further I will suggest you to download the pre-trained model from ESPNET and with slight modifications and fine-tuning, you should be able to get good results.
I used this model to test, and the following exception occurred:
File "/home/Documents/ETE/test.py", line 105, in main
model.load_state_dict(checkpoint['model'])
File "/home/.local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 777, in load_state_dict
self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for E2E:
Can you send the entire log of the exception? I believe the log you mentioned is truncated.
log:
Starting Testing
Traceback (most recent call last):
File "main.py", line 61, in
I have made the corresponding changes in test.py. I haven't tested it on my machine, but it should work.
The problem is that DataParallel module changes the name of the layers and when you try to load it on PC without the module part, pytorch is not able to map the weights. A simple way to solve this is to remove the "module." part from the keys of the state dictionary
Now can run the code normally. Thank you very much.
Glad I could help!
Hi. The training is very time-consuming, so could provide the pretrained model, if it is available? Thank you very much!