()
| 27 | |
| 28 | |
| 29 | def check_torch_version(): |
| 30 | torch_minimum_version = '1.8.0' |
| 31 | torch_version = digit_version(torch.__version__) |
| 32 | |
| 33 | assert (torch_version >= digit_version(torch_minimum_version)), \ |
| 34 | f'Torch=={torch.__version__} is not support for converting to ' \ |
| 35 | f'torchscript. Please install pytorch>={torch_minimum_version}.' |
| 36 | |
| 37 | |
| 38 | def _convert_batchnorm(module): |
no test coverage detected