hyrax.verbs.to_onnx

Attributes

Classes

Export the model to ONNX format

Module Contents

logger[source]

class ToOnnx(config)[source]

Bases: hyrax.verbs.verb_registry.Verb

Export the model to ONNX format

__init__()[source]

Overall initialization for all verbs that saves the config

cli_name = 'to_onnx'[source]

add_parser_kwargs[source]

static setup_parser(parser)[source]: Setup parser for ONNX export verb

run_cli(args=None)[source]: Run the ONNX export verb from the CLI

run(input_model_directory: str = None)[source]: Export the model to ONNX format and save it to the specified path.