ignition-monorepo

ignition-monorepo / backend-onnx/src / saveForOnnxExport

Function: saveForOnnxExport()

saveForOnnxExport(model, outputDir, onnxOutputPath?, opset?): Promise<ExportResult>

Defined in: backend-onnx/src/exporter.ts:54 

Saves a TF.js LayersModel to outputDir and returns the Python conversion script.

Parameters

model

A TF.js tf.LayersModel (e.g. dqnAgent.getModel())

save

(url) => Promise<unknown>

outputDir

string

Directory path where the TF.js model will be saved

onnxOutputPath?

string

Desired output path for the final .onnx file (default: <outputDir>.onnx)

opset?

number = 13

Returns

Promise<ExportResult>