Giters

tianyic / only_train_once_personal_footprint

OTOv1-v3, NeurIPS, ICLR, TMLR, DNN Training, Compression, Structured Pruning, Erasing Operators, CNN, Diffusion, LLM

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ONNX opset setting

unique-null opened this issue · comments

commented

Hi @tianyic ,

I run OTO on my project. However, there is an error which describe "RuntimeError: Exporting the operator im2col to ONNX opset version 9 is not supported. Support for this operator was added in version 11, try exporting with this version.".

I use pytorch 1.10.1, which supports opset=13. However, I didn't find the interface for setting the opset. How can I fix the problem.

thx.

commented

Thanks for reaching out. We will add interface to support opt version into the next release.

For a hotfix, I suggest to look into the compression.py

There exist the onnx export functions where I think you could set the opt version there.

Hope the above help.