adetuire1 commited on
Commit
8084b7e
·
verified ·
1 Parent(s): f95bf2c

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +6 -9
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
- **Base model:** `gpt2`
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.