Open jlamperez opened 2 years ago
Hi @jlamperez ,
tvm
discourage the use of aten::copy_
, you can check this discussion: https://github.com/apache/tvm/pull/7231#issuecomment-756693366 for more details.aten::silu_
, you can use something like below to resolve this ops:
https://github.com/Megvii-BaseDetection/YOLOX/blob/902b372be820a738ee95c55da2a957c5c2161536/yolox/models/network_blocks.py#L9-L14BTW, I've done some experiments on YOLOv5
with tvm
here, and the differences between this two models are relatively insignificant, so I believe with some minor changes we can also use tvm
on YOLOX
.
Hi,
Thank you for this nice repo.
I am trying to export pytorch model to TVM with this snippet of code
but I am having the next problem
How these operators (silu and copy) can be implemented in TVM?
Best regards!