onnx_diagnostic.tasks

modules

• onnx_diagnostic.tasks.automatic_speech_recognition
• onnx_diagnostic.tasks.fill_mask
• onnx_diagnostic.tasks.image_classification
• onnx_diagnostic.export.image_text_to_text
• onnx_diagnostic.tasks.sentence_similarity
• onnx_diagnostic.tasks.text_classification
• onnx_diagnostic.tasks.text_generation
• onnx_diagnostic.tasks.text2text_generation
• onnx_diagnostic.tasks.zero_shot_image_classification

onnx_diagnostic.tasks.random_input_kwargs(config: Any, task: str) → Tuple[Dict[str, Any], Callable]

Inputs kwargs. If the configuration is None, the function selects typical dimensions. It returns parameters and a function. The function creates dummy inputs if it receives the parameters returned as a first result.

onnx_diagnostic.tasks.reduce_model_config(config: Any, task: str) → Dict[str, Any]

Reduces a model size.

onnx_diagnostic.tasks.supported_tasks() → List[str]

Returns the list of supported tasks.