Returns True when torch's version is at least `min_version`.
(min_version: str)
| 163 | |
| 164 | |
| 165 | def min_torch_version(min_version: str) -> bool: |
| 166 | """ |
| 167 | Returns True when torch's version is at least `min_version`. |
| 168 | """ |
| 169 | try: |
| 170 | import torch |
| 171 | except ImportError: |
| 172 | return False |
| 173 | |
| 174 | installed_version = version.parse(torch.__version__.split("+")[0]) |
| 175 | min_version = version.parse(min_version) |
| 176 | return installed_version >= min_version |
| 177 | |
| 178 | |
| 179 | def has_dynamic_axes(onnx_model): |
no outgoing calls
no test coverage detected