| language: en | |
| library_name: transformers | |
| pipeline_tag: text-generation | |
| tags: | |
| - gpt2 | |
| - causal-lm | |
| - imdb | |
| - fine-tuning | |
| datasets: | |
| - imdb | |
| license: mit | |
| base_model: gpt2 | |
| widget: | |
| - text: Once upon a time | |
| - text: The movie was | |
| # GPT-2 IMDB Fine-tuned | |
| **Base model:** `gpt2` | |
| **Dataset:** IMDB (train + unsupervised) | |
| **Task:** Causal LM | |
| > For learning/demo use only. | |