nshepperd/gpt-2

Image GPT Training

cellininicholas opened this issue · 0 comments

Hey!
I was looking into re-training the pre-trained models in Image GPT.
Since the project was forked from GPT2, I was wondering if your train.py might work for Image GPT. (Since I can't find any documentation in the Image GPT repo for training.)

Anyone have any insights?
Thanks in advance! :)