shreydan/VisionGPT2

Combining ViT and GPT-2 for image captioning. Trained on MS-COCO. The model was implemented mostly from scratch.

语言：Jupyter Notebook · GitHub 星标：49 · Fork：3

话题：gpt · image-captioning · multimodal · pytorch · transformers · vit

在 GitHub 上查看原仓库