sayakpaul/probing-vits

Dino Base models in saved model format

keertika-11 opened this issue · 1 comments

how can the dino base model checkpoints ( Feature Extractor) in pytorch or onnx be converted into Saved Model Format ( Tensorflow)? which will help in finetuning the dino model

You could discard the classification head from ViTDINOBase (shown in this notebook) and follow the rest of the process.