How to test your model after training on my own dataset?

codertimo / BERT-pytorch

Google AI 2018 BERT pytorch implementation

Apache License 2.0

6.19k stars 1.3k forks source link

How to test your model after training on my own dataset? #57

Open riktimmondal opened 5 years ago

riktimmondal commented 5 years ago

I am using for next sent. gen so while it has stored model in .ep* format but how to run my test dataset using those models.

Thanks

jbmaxwell commented 5 years ago

I have a similar question. My understanding is that the output should be a distribution over tokens for each predicted (i.e., "next") output, but I'm wondering how to get an actual string result?... Did you manage to get your next sentence generation working?