Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -1,24 +1,21 @@
|
|
1 |
---
|
2 |
language: en
|
3 |
-
library_name: transformers
|
4 |
pipeline_tag: text-generation
|
|
|
|
|
5 |
tags:
|
6 |
- gpt2
|
7 |
-
- causal-lm
|
8 |
- imdb
|
|
|
9 |
- fine-tuning
|
|
|
10 |
datasets:
|
11 |
- imdb
|
12 |
-
license: mit
|
13 |
-
base_model: gpt2
|
14 |
widget:
|
15 |
-
- text: Once upon a time
|
16 |
- text: The movie was
|
|
|
17 |
---
|
18 |
|
19 |
# GPT-2 IMDB Fine-tuned
|
20 |
|
21 |
-
|
22 |
-
**Dataset:** IMDB (train + unsupervised)
|
23 |
-
**Task:** Causal LM
|
24 |
-
> For learning/demo use only.
|
|
|
1 |
---
|
2 |
language: en
|
|
|
3 |
pipeline_tag: text-generation
|
4 |
+
library_name: transformers
|
5 |
+
license: mit
|
6 |
tags:
|
7 |
- gpt2
|
|
|
8 |
- imdb
|
9 |
+
- causal-lm
|
10 |
- fine-tuning
|
11 |
+
base_model: gpt2
|
12 |
datasets:
|
13 |
- imdb
|
|
|
|
|
14 |
widget:
|
|
|
15 |
- text: The movie was
|
16 |
+
- text: Once upon a time
|
17 |
---
|
18 |
|
19 |
# GPT-2 IMDB Fine-tuned
|
20 |
|
21 |
+
Finetuned GPT-2 on IMDB (train + unsupervised) for causal language modeling.
|
|
|
|
|
|