Since torch functions often allow for models to either be from disk as fpath or from a loaded object, we provide a similar option here. Arguments can either be a path on disk or from model itself. Args: model (Union[str, torch.Model]): Location of the model as f
(
self,
default_converter: ModelFileConverter = None,
network_metadata: NetworkMetadata = None,
)
| 158 | """ |
| 159 | |
| 160 | def __init__( |
| 161 | self, |
| 162 | default_converter: ModelFileConverter = None, |
| 163 | network_metadata: NetworkMetadata = None, |
| 164 | ): |
| 165 | """ |
| 166 | Since torch functions often allow for models to either be from disk as fpath or from a loaded object, |
| 167 | we provide a similar option here. Arguments can either be a path on disk or from model itself. |
| 168 | |
| 169 | Args: |
| 170 | model (Union[str, torch.Model]): Location of the model as fpath OR loaded torch.Model object. |
| 171 | """ |
| 172 | if default_converter is not None: |
| 173 | self.default_converter = default_converter() |
| 174 | else: |
| 175 | self.default_converter = NullConverter() |
| 176 | |
| 177 | self.network_metadata = network_metadata |
| 178 | |
| 179 | def as_torch_model( |
| 180 | self, |
no test coverage detected