Downloads · 30 days
8
16% of all-time downloads
gpt2-dropout
ao9000/gpt2-dropout
gpt2-dropout is a text generation model from ao9000. Use it when you need the model to write or continue text. It is set up for transformers.
- Dataset: Wikipedia-10 - Target modules = ["query", "value"]
All-time downloads
49
Public
Parameters
124M
249 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors249 MB · 100%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- gpt2
- Access
- Public
- Created
- Jul 22, 2024
- Updated
- Jul 22, 2024
- SHA
- 6ff99c90
Try a prompt
Model card
From the Hugging Face model README
Lora finetuning on Wikipedia-10, applying increased dropout for all layers
- Dataset: Wikipedia-10
- Target modules = ["query", "value"]
{
"epoch": 0.22326645805206993,
"total_flos": 3.3909236563968e+16,
"train_loss": 3.594551098823547,
"train_runtime": 5469.927,
"train_samples": 286653,
"train_samples_per_second": 11.7,
"train_steps_per_second": 0.366
}
Training script: https://github.com/ao9000/bias-bench/blob/main/experiments/run_clm.py
- Task
- Text Generation
- Library
- transformers
- Type
- gpt2
- Created
- Jul 22, 2024
- Updated
- Jul 22, 2024