Please add these changes to your train.py

prakhar21 / TextAugmentation-GPT2

Fine-tuned pre-trained GPT2 for custom topic specific text generation. Such system can be used for Text Augmentation.

MIT License

187 stars 43 forks source link

Closed sayantanbasu05 closed 3 years ago

sayantanbasu05 commented 4 years ago

Now the parameter should be passed like this, --data_file data/SMSSpamCollection

The original train.py was not running on my colab. I made a few changes. This file runs smoothly on google colab and may help many.