google-research / scenic

Scenic: A Jax Library for Computer Vision Research and Beyond
Apache License 2.0
3.28k stars 428 forks source link

Save the model (Owl-ViT) convert it and use it for custom pipeline #1073

Open Aki1608 opened 4 months ago

Aki1608 commented 4 months ago

Hi all,

This is more of a question than an issue. I want to train the model (Owl-ViT) on custom dataset from checkpoint ("google/owlvit-base-patch32" or clip_vit) and want to use it in custom pipeline. For that I want to convert this model into ".engine" file. Is there any way of saving the model in a particular format and convert it to ONNX format?

Thank you.

Best regards, Akash.