pytorch-openai-transformer-lm
pytorch-openai-transformer-lm copied to clipboard
Running on new dataset similar to rocstories
Hi all,
I am trying to train a new dataset with a similar structure to rocstories. It has a story part, 2 options and one correct option. I just added a new function in datasets.py but this is not enough. I am not able to train. Has anyone done that and can provide me with some suggestions?
Thanks in advance.