Skip to content

Export ONNX model during training #5142

@leoll2

Description

@leoll2

Extend the training job export_model function to also export as ONNX (model.onnx). Also update store_model_artifacts accordingly.

Note: in the future, we want to create the ONNX model on-demand rather than by default

Metadata

Metadata

Assignees

Labels

Geti Tune BackendIssues related to Geti Tune backend

Type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions