Compile CLIP with the DFC

[lukasOST]

I have to withdraw my statement. I didn’t notice I was working with the changes I mentioned earlier in this forum.
[my fix]

I still get the error:

spatial_reshape_sizes = [output_shape[1], output_shape[2]]
IndexError: list index out of range

@nina-vilela, do you have another idea where my error is?

[nina-vilela]

Please try again from a fresh clean venv, and make sure that you install clip the recommended way:

pip install git+https://github.com/openai/CLIP.git

Also. please try exactly this snippet:

@nina-vilela

import torch
import clip
from torch.autograd import Variable

vision_arch = "RN50x4"
device = "cpu"
model, preprocess = clip.load("RN50x4", device=device)

vit = model.visual

dummy_input = Variable(torch.randn(1, 3, 288, 288, device="cpu"))

# ==================== #
# Export Model as ONNX #
# ==================== #
vit.eval()
torch.onnx.export(vit,          # model being run 
         dummy_input,                 # model input (or a tuple for multiple inputs) 
         "clip.onnx",       # where to save the model  
         opset_version=12,      # the ONNX version to export the model to
         do_constant_folding=True,          # whether to execute constant folding for optimization 
         )

[lukasOST]

Hey @nina-vilela,
I made a fresh venv and installed the following packages:

pip install torch
pip install torch.onnx
pip install git+https://github.com/openai/CLIP.git

Then i executed your Code but i again had to adjust the opset_version = 14`
The result is this: