tiny-gpt2 (ONNX)
This is an ONNX version of taufeeque/tiny-gpt2. It was automatically converted and uploaded using this Hugging Face Space.
Usage with Transformers.js
See the pipeline documentation for text-generation: https://huggingface.co/docs/transformers.js/api/pipelines#module_pipelines.TextGenerationPipeline
output_tiny
This model is a fine-tuned version of gpt2_tiny_random on the wikitext wikitext-103-v1 dataset. It achieves the following results on the evaluation set:
- Loss: 5.3359
- Accuracy: 0.2133
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 0.0001
- train_batch_size: 32
- eval_batch_size: 64
- seed: 42
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- training_steps: 50000
Training results
Framework versions
- Transformers 4.25.1
- Pytorch 1.13.1
- Datasets 2.8.0
- Tokenizers 0.13.2
- Downloads last month
- 7
Model tree for onnx-community/tiny-gpt2-ONNX
Base model
taufeeque/tiny-gpt2Dataset used to train onnx-community/tiny-gpt2-ONNX
Evaluation results
- Accuracy on wikitext wikitext-103-v1self-reported0.213