search

rish-16 / gpt2client

✍🏻 gpt2-client: Easy-to-use TensorFlow Wrapper for GPT-2 117M, 345M, 774M, and 1.5B Transformer Models 🤖 📝
MIT License
372 stars 74 forks source link

How do I resume training from checkpoint #26

Closed viveklse closed 4 years ago

viveklse commented 4 years ago

Describe the bug How do I resume training from a checkpoint? Also, how do I save the trained model to my drive?

I am asking this because I know this package is downloading gpt2simple (Successfully installed gpt-2-simple-0.6 gpt2-client-2.0 regex-2019.8.19 toposort-1.5) and using the following: gpt2.copy_checkpoint_to_gdrive(run_name=model-name') you can save your trained model to your drive.

Thanks

rish-16 commented 4 years ago

Hey, for now, the package does not have a "resume from checkpoint" kind of method. Will put it on the list, though :)