Downloads · 30 days
18
2% of all-time downloads
recipe-nlg-gpt2-train11_14
pratultandon/recipe-nlg-gpt2-train11_14
recipe-nlg-gpt2-train11_14 is a text generation model from pratultandon. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as mit.
should probably proofread and complete it, then remove this comment. --
All-time downloads
860
Public
Repo size
5.1 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin510 MB · 99%
At a glance
- Task
- Text Generation
- Library
- transformers
- License
- mit
- Model type
- gpt2
- Access
- Public
- Created
- Nov 14, 2022
- Updated
- Nov 16, 2022
- SHA
- 33ba28a6
Try a prompt
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. -->
recipe-nlg-gpt2-train11_14
This model is a fine-tuned version of gpt2 on the Recipe-NLG dataset.
Model description
TEST MODEL - LESS THAN 0.10 EPOCHS OF TRAINING COMPLETED
Intended uses & limitations
Experimenting with GPT-2 for recipe generation.
Training and evaluation data
The RecipeNLG(https://huggingface.co/mbien/recipenlg/) dataset was used for this task.
5% of the dataset was held out for evaluation.
Training procedure
RTX 3090 was used on Vast.AI, training took about 14 hours with a batch size of 8, and f16 enabled.
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 5e-05
- train_batch_size: 8
- eval_batch_size: 8
- seed: 42
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- lr_scheduler_warmup_steps: 200
- num_epochs: 1
- mixed_precision_training: Native AMP
Framework versions
- Transformers 4.24.0
- Pytorch 1.13.0
- Datasets 2.6.1
- Tokenizers 0.13.2
- Task
- Text Generation
- Library
- transformers
- Type
- gpt2
- License
- mit
- Created
- Nov 14, 2022
- Updated
- Nov 16, 2022