vivit-weights-converter

Script to convert ViViT weights from Flax to Pytorch.

Use

Before use, please have torch and flax installed.

python convert_vivit_flax_to_pytorch.py --flax_model PATH_TO_FLAX_CHECKPOINT --output_model_name TORCH_MODEL_NAME

Add flag --classification_head to include classification head weights.