Skip to content

PyTorch Vision Transformer (ViT) for Image Classification

Compare
Choose a tag to compare
@quic-bharathr quic-bharathr released this 09 Dec 09:20
· 14 commits to develop since this release

This releases PyTorch W8A8 optimized weight and quantization config file for vision transformer model (ViT) for image classification tasks using ImageNet dataset