Hey,
I plan to export a torchscript graph represantation by using trace(...) command, and then using trace.graph(). after that, I plan to apply some optimisations for the given graph and then I want to revert the result back to ScriptModule class.
I am familiar with: torch._C._create_function_from_graph('forward', graph)
but it returns ScriptFunction class. I would like to use it as a module (ScriptModule class).
I need it as a module class because I need to be able to access its parameters by parameters() method.
thanks, Omer.