Time to train the model????? Please tell the estimate time.

daniel-kukiela / nmt-chatbot

NMT Chatbot

GNU General Public License v3.0

387 stars 214 forks source link

Time to train the model????? Please tell the estimate time. #151

Open SkullEnemyX opened 4 years ago

SkullEnemyX commented 4 years ago

I have created a dataset from the reddit 2015 publicly available dataset from the SentDex video and prepared a 50 million pairs of comments and replies. Now, I have a GPU GTX 1050Ti with 4gb ram, and the configuration of my vocab size is 20K. Can someone tell me the estimated time to train the model?

kaljitism commented 4 years ago

It will take days

Nathan-Chell commented 4 years ago

Also, you won't want a vocab size of 20k. That's very small. What is the size of your model? Num layers and num units?

SkullEnemyX commented 4 years ago

It took me 8 hours to train a model of vocab size 20k and 200K million pairs of comments.