Specify the number of epochs when finetuning gpt2client

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

Specify the number of epochs when finetuning gpt2client #34

Closed BrianMwangi21 closed 3 years ago

BrianMwangi21 commented 4 years ago

How many epochs does it take the gpt2client to fine tune and can one specify explicitly ?

I have been running on Colab and now it is at 177 and I am wondering how long I have to go yet. Thanks

rish-16 commented 3 years ago

Hey there,

The finetuning feature of gpt2-client uses a third-party finetuning library called gpt2-simple. All the details on finetuning can be found there. So sorry for the delayed response – I haven't had the time to work on this project for a while.

Appreciate your patience!