Closed mshr-h closed 3 weeks ago
The error message shows that operation Hardswish
is not supported in the fx converter, would you mind sending a pr to add it in https://github.com/apache/tvm/blob/main/python/tvm/relax/frontend/torch/fx_translator.py?
Thank you for your comment! Yes, I'm implementing hardswish support on the torch frontend (fx_translater.py). Initial implementation works fine. After clearing up my code, I'll send the PR.
Cannot import mobilenet_v3 because Hardswish and Hardsigmoid are not supported by Relax. I'll try to fix it.
TODOs
Expected behavior
mobilenet_v3_small and mobilenet_v3_small can be imported with
from_fx
.Actual behavior
Got the below error message when I executed the repro.
Environment
OS: Ubuntu 22.04 LTS on WSL2 TVM: 0e622e140c2df8c5ab88e27ee4e90254cddb80ce PyTorch: 2.3.0 Torchvision: 0.18.0
Steps to reproduce
Triage
Please refer to the list of label tags here to find the relevant tags and add them below in a bullet format (example below).
cc @junrushao