search

google-research / vision_transformer

Apache License 2.0
10.48k stars 1.3k forks source link

can export the pretrained model to onnx or pytorch? #293

Open alanguo1234 opened 11 months ago

alanguo1234 commented 11 months ago

Hi my compiler support onnx pytorch model at frontend, how to export the Vision Transformer pretrained model to onnx or pytorch ? if can provide sample, it would be even better， thanks

andsteing commented 11 months ago

We don't provide PyTorch implementations, but all models are supported in the fantastic timm library.

alanguo1234 commented 11 months ago

We don't provide PyTorch implementations, but all models are supported in the fantastic timm library.

thanks for your quickly reply, by the way, can it export to tensorflow or tensorflow lite format