-
Notifications
You must be signed in to change notification settings - Fork 3.5k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Bug] YOLOX Autotune error - RuntimeError: Invalid type of axis: <class 'tvm.tir.expr.SizeVar'> #11780
Comments
Does it compile and run without autotvm? I think maybe the ONNX frontend is generating a conv2d op whose input width and height are dynamic, when they shouldn't be. I've seen the same issue for MaskRCNN when imported via ONNX. If that's the case, this is not an autotvm problem. I can take a look if you can post your |
@alexandrepires5 Can you check the model url?
That would be a bug in the frontend, so it is not easy to fix. But given that things work outside of autotvm, this is probably not a problem. |
@masahi yes, sorry, I was distracted and reverted the markdown haha. Can you download the model without any issues? I will not be able to check this with more detail until this friday. By the end of the week, I can also give it a look and try solutions if any can be provided. |
With the latest
So maybe v0.8.0 had a issue that has been fixed now. Can you try our latest branch? |
You are right @masahi, with the latest version, it auto tunes. I guess there was some issue in 0.8.0 that has been fixed recently. For everybody else wondering, I installed using tlcpack, the version I have is this one: tlcpack-nightly==0.9.dev1679+g59fb4219f So, from that version, yolox should be able to autotune without any workaround. I think we can close this issue. |
Hello there,
So, I have been having this issue trying to autotune an onnx based Yolox nano model(416x416). I am playing around with the autotune configuration using the python API, but I get this runtime error:
RuntimeError: Invalid type of axis: <class 'tvm.tir.expr.SizeVar'>
The code I have is the following:
The following are my logs:
My version of TVM is 0.8.0, I have a ryzen 2400g with a GTX 1070.
Is there something I am doing wrong or it this a TVM issue? Issue #10042 seemed to have the same problem, but I want to avoid using the workaround. Is there any clean way of fixing this?
The text was updated successfully, but these errors were encountered: