GPT NEO-For COVID-19 Question Answering

I am working on a NLP project. It is about question answering system of COVID-19 where i am using covid-19 open resource dataset from Kaggle.

I am using GPT-Neo for training my data. Initially I worked with a very small portion of my data which by default took only 3 epoch. I am trying to increase the epoch size into 50 or more. How will i be able to do that? In gpt mode. Here is the ss of my code.