hyrax.verbs.to_onnx

hyrax.verbs.to_onnx#

Attributes#

logger

Classes#

ToOnnx

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]#
description = 'Export model to ONNX format.'[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.