Skip to content

Function: modelGraphFromTorchFx()

ts
function modelGraphFromTorchFx(nodes, opts?): ModelGraph;

Defined in: ml/model.ts:211

Build a ModelGraph from traced torch.fx nodes (keeps skip connections).

Parameters

ParameterType
nodesTorchFxNode[]
optsAdapterOptions

Returns

ModelGraph

MIT licensed. WebGL2 required.