search

reppy4620 / Dialog

A PyTorch Implementation of japanese chatbot using BERT and Transformer's decoder
MIT License
72 stars 29 forks source link

training error #5

Closed kahnlzz closed 4 years ago

kahnlzz commented 5 years ago

Thank you for the great work. I followed the usage guide in the github but some errors occurred. Any code not changed. The following error occurred when copying pretrained model to './pretrained' folder and executing 'python main.py'. "OSError: file ./pretrained/config.json not found" 'config.json' file was not found in your github. When I copied the 'config.json' from original Bert, it was cleared. After that, I got an error "OSError: file ./pretrained/pytorch_model.bin not found" Please give me some tips to solve it.

Thanks in advance. Seungkwon.

reppy4620 commented 5 years ago

Sorry, I forgot to share about pretrained models.

In fact, for preparing "pytorch.bin", you have to convert original tf model to pytorch model using huggingface's convert script. but it's too much bother to convert model. Therefore I opened my "pretrained" folder to the public. This is the link.

Thank you for letting me know.

kahnlzz commented 4 years ago

Thank you for the kind response