facebookresearch / dino

PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
Apache License 2.0
6.06k stars 885 forks source link

How to convert to onnx #276

Open Ochre-amber opened 2 months ago

Ochre-amber commented 2 months ago

Hello, I noticed that the pre-trained onnx file is provided, but I am not very successful in converting my own model, how do I convert the dino model to onnx format?

Ochre-amber commented 2 months ago

Has anyone ever had the same problem as me?

wangh09 commented 2 hours ago

is there a specific error that you encountered?