Skip to content

Latest commit

 

History

History
20 lines (14 loc) · 804 Bytes

README.md

File metadata and controls

20 lines (14 loc) · 804 Bytes
Raw

Torch ViT

Quickstart

Install with pip install torch_vit and load a pretrained ViT with:

from torch_vit import ViT
model = ViT('B_16_imagenet1k', pretrained=True)

Overview

This repository contains an PyTorch reimplementation of the Visual Transformer architecture from Google, along with pre-trained models and and wrapper modules and examples. Some other variations will developed in next releases

The goal of this implementation is to be simple, highly extensible, and easy to integrate into your own projects.

  • At the moment, you can easily:
    • Loading Pretrained ViT Models
    • Evaluate on ImageClassification or Custom data
    • Finetune ViT on Custom Dataset