vit-base-ipu / README.md
jimypbr's picture
Create README.md
352e109
|
raw
history blame
488 Bytes

ViT Base model IPU config

This model contains just the IPUConfig files for running the ViT base model (e.g. vit-base-patch16-224-in21k or deit-base-patch16-384) on Graphcore IPUs.

This model contains no model weights, only an IPUConfig.

Usage

from optimum.graphcore import IPUConfig
ipu_config = IPUConfig.from_pretrained("Graphcore/vit-base-ipu")