Downloads · 30 days
17
4% of all-time downloads
openai_gpt2_med_convai
SSahas/openai_gpt2_med_convai
openai_gpt2_med_convai is a text generation model from SSahas. Use it when you need the model to write or continue text. It is set up for peft. The card lists the license as mit.
should probably proofread and complete it, then remove this comment. --
All-time downloads
401
Public
Repo size
425 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors425 MB · 99%
At a glance
- Task
- Text Generation
- Library
- peft
- License
- mit
- Access
- Public
- Created
- Mar 17, 2024
- Updated
- Apr 2, 2024
- SHA
- d2aed5f4
Try a prompt
Base models
Datasets
Model card
From the Hugging Face model README
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
should probably proofread and complete it, then remove this comment. -->
openai_gpt
This model is a fine-tuned version of openai-community/gpt2-medium on the daily_dialog dataset.
Model description
Intended uses & limitations
The model can be used for friendly conversations
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 0.0025
- train_batch_size: 3
- eval_batch_size: 8
- seed: 42
- gradient_accumulation_steps: 2
- total_train_batch_size: 6
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: cosine
- lr_scheduler_warmup_ratio: 0.03
- training_steps: 200
Training results
- Training Loss : 2.113900
Framework versions
- PEFT 0.9.1.dev0
- Transformers 4.38.2
- Pytorch 2.0.0
- Datasets 2.18.0
- Tokenizers 0.15.2
- Task
- Text Generation
- Library
- peft
- License
- mit
- Created
- Mar 17, 2024
- Updated
- Apr 2, 2024