Dino Base models in saved model format
keertika-11 opened this issue · 1 comments
keertika-11 commented
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
sayakpaul commented
You could discard the classification head from ViTDINOBase
(shown in this notebook) and follow the rest of the process.