bryanlimy / tf2-transformer-chatbot

Transformer Chatbot in TensorFlow 2 with TPU support.
MIT License
130 stars 58 forks source link

Can we provide a trained model with the code base and also separate nb file on how to load the existing model to use?? #27

Open robert-cui1 opened 1 year ago

robert-cui1 commented 1 year ago

Thanks for the code base, it served me well:) Just wondering whether it's possible to have a trained model with the code base, which will save some enthusiasts like me lots of time? I took me about 6 hours on my mac. It would be also better if there is a separate Jupyter nb file, which is specifically for users, who don't want to train the model, but straight to load and use an existing model to play with. I know there is some code for loading and using the model, but the code is part of the whole process, not strictly to load and use a model without dependency on other pieces of code. Thanks:)