Thanks a lot for replying to me and my second question is that I want to train the gpt-2-simple with custom dictionary where key is word and value i is meaning. Could you suggest me what should the size of the file for training, I would like to use 117 M for such. Does there is some restriction on how to make the dataset.

My other question is that during prediction phase I would like to input word and model will predict the meaning of it. Can you help me out for this.

Thanks and with regards

Sanpreet Singh