how to transform torch.fx.GraphModule to nn.module?

ENOT-AutoDL / onnx2torch

Convert ONNX models to PyTorch.

Apache License 2.0

587 stars 69 forks source link

how to transform torch.fx.GraphModule to nn.module? #216

Closed 2002liush closed 1 month ago

2002liush commented 4 months ago

After convert we get a model belongs to torch.fx.GraphModule, but we want to get a model belongs to nn.module. How to transform a torch.fx.GraphModule to nn.module?

senysenyseny16 commented 1 month ago

hi @2002liush,

At the moment, this is not possible.