sidsingla opened a new issue, #17842:
URL: https://github.com/apache/tvm/issues/17842
Hi,
Here is the Traceback:
Traceback (most recent call last):
```
File
"/home/sidharth_modiface_com/looks_factory_tvm/looks-factory-transfer/convert_to_tvm.py",
line 93, in <module>
mod = from_fx(
^^^^^^^^
File
"/opt/conda/envs/py311-looks-factory-tvm/lib/python3.11/site-packages/tvm/relax/frontend/torch/fx_translator.py",
line 1064, in from_fx
return TorchFXImporter().from_fx(
^^^^^^^^^^^^^^^^^^^^^^^^^^
File
"/opt/conda/envs/py311-looks-factory-tvm/lib/python3.11/site-packages/tvm/relax/frontend/torch/fx_translator.py",
line 935, in from_fx
assert (
AssertionError: Unsupported module type <class
'torch.nn.modules.instancenorm.InstanceNorm2d'>
```
Code snippet:
```
mod = from_fx(
fx_graph,
[((1, 3, 256, 256), "float32"), ((1, 1, 256, 256), "float32")],
keep_params_as_input=True,
)
```
Model is mostly based on pix2pix GAN model.
Thanks!
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
