Keeps track of ONNX model file. Does not support loading into memory. Only reads and writes to disk. Args: model (str): Location of the model as fpath OR loaded torch.Model object.
(
self,
model: str,
default_converter: ModelFileConverter = None,
network_metadata: NetworkMetadata = None,
)
| 361 | |
| 362 | class ONNXModelFile(NNModelFile): |
| 363 | def __init__( |
| 364 | self, |
| 365 | model: str, |
| 366 | default_converter: ModelFileConverter = None, |
| 367 | network_metadata: NetworkMetadata = None, |
| 368 | ): |
| 369 | """ |
| 370 | Keeps track of ONNX model file. Does not support loading into memory. Only reads and writes to disk. |
| 371 | |
| 372 | Args: |
| 373 | model (str): Location of the model as fpath OR loaded torch.Model object. |
| 374 | """ |
| 375 | super().__init__(default_converter, network_metadata) |
| 376 | self.fpath = model |
| 377 | |
| 378 | def as_onnx_model( |
| 379 | self, |