tarekziade/distilvit
By tarekziade · ↓ 16 · ♥ 2 · Updated 2024-03-18
What's in this repo
This page lists the files in tarekziade/distilvit, including decoder_model.onnx, decoder_model_merged.onnx, decoder_model_merged_quantized.onnx and more.
- 8 ONNX models (.onnx files)
- 21 files, including 8 .json, 8 .onnx, 1 .md, 1 .txt, 1 .safetensors, 1 .bin
- Last updated 2024-03-18
ONNX files
| File | Size | Download |
|---|---|---|
| onnx/decoder_model.onnx | — | Download |
| onnx/decoder_model_merged.onnx | — | Download |
| onnx/decoder_model_merged_quantized.onnx | — | Download |
| onnx/decoder_model_quantized.onnx | — | Download |
| onnx/decoder_with_past_model.onnx | — | Download |
| onnx/decoder_with_past_model_quantized.onnx | — | Download |
| onnx/encoder_model.onnx | — | Download |
| onnx/encoder_model_quantized.onnx | — | Download |
How to use
These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.
README
This model is a variation of https://huggingface.co/nlpconnect/vit-gpt2-image-captioning
- Read the blog post here https://ziade.org/2024/03/17/distilvit-image-captioning-model
- The training code is here: https://github.com/tarekziade/distilvit
Results after after 3 epochs (and ~45 hours of training)
- eval_loss: 0.19939416646957397
- eval_rouge1: 43.006
- eval_rouge2: 16.9939
- eval_rougeL: 38.8923
- eval_rougeLsum: 38.8877
- eval_gen_len: 11.327256736227712
- eval_runtime: 1816.5255
- eval_samples_per_second: 13.77
- eval_steps_per_second': 1.721
- train_runtime: 46263.3695
- train_samples_per_second: 38.373
- train_steps_per_second: 4.797
- train_loss: 0.05974134062104816
Model info
image-to-text
transformers
transformers
onnx
safetensors
vision-encoder-decoder
image-text-to-text
image-to-text
image-captioning
base_model:distilbert/distilgpt2
base_model:quantized:distilbert/distilgpt2
license:apache-2.0
endpoints_compatible
region:us
↓ 16
♥ 2
Created: 2024-03-16
Updated: 2024-03-18
View on Hugging FaceFiles (21)
.gitattributes
README.md
config.json
generation_config.json
merges.txt
model.safetensors
onnx/decoder_model.onnx
ONNX
onnx/decoder_model_merged.onnx
ONNX
onnx/decoder_model_quantized.onnx
ONNX
onnx/decoder_with_past_model.onnx
ONNX
onnx/decoder_with_past_model_quantized.onnx
ONNX
onnx/encoder_model.onnx
ONNX
onnx/encoder_model_quantized.onnx
ONNX
preprocessor_config.json
quantize_config.json
special_tokens_map.json
tokenizer.json
tokenizer_config.json
training_args.bin
vocab.json