onnx_diagnostic.torch_models.hghub.hub_data

onnx_diagnostic.torch_models.hghub.hub_data.load_architecture_task() Dict[str, str][source]

Returns a dictionary mapping architecture to task.

import pprint from onnx_diagnostic.torch_models.hghub.hub_data import load_architecture_task pprint.pprint(load_architecture_task())

onnx_diagnostic.torch_models.hghub.hub_data.load_models_testing() List[str][source]

Returns model ids for testing.