Skip to content

model_wrapper

torch_to_nnef.model_wrapper

Wrap model to bypass limitation of torch_to_nnef internals.

ie: Cases where inputs or outputs of a model contains tuples

WrapStructIO

WrapStructIO(model: nn.Module, input_infos, output_infos)

Bases: Module

Once traced it should be nop in final graph.