Has anyone been able to successfully convert the timm+tome model to onnx?

facebookresearch / ToMe

A method to increase the speed and lower the memory footprint of existing vision transformers.

Other

931 stars 67 forks source link

Has anyone been able to successfully convert the timm+tome model to onnx? #32

Closed Roman54228 closed 1 year ago

Roman54228 commented 1 year ago

Facing difficulties with onnx opset vesrions and operators used in this implementation, cant convert timm model to onnx. Is there a way to do it maybe in other way?

dbolya commented 1 year ago

You might be interested in this PR for tomesd: https://github.com/dbolya/tomesd/pull/43

Roman54228 commented 1 year ago

Thanks you a lot!!! It worked