| | import warnings |
| |
|
| | import torch |
| |
|
| | from annotator.uniformer.mmcv.utils import digit_version |
| |
|
| |
|
| | def is_jit_tracing() -> bool: |
| | if (torch.__version__ != 'parrots' |
| | and digit_version(torch.__version__) >= digit_version('1.6.0')): |
| | on_trace = torch.jit.is_tracing() |
| | |
| | |
| | if isinstance(on_trace, bool): |
| | return on_trace |
| | else: |
| | return torch._C._is_tracing() |
| | else: |
| | warnings.warn( |
| | 'torch.jit.is_tracing is only supported after v1.6.0. ' |
| | 'Therefore is_tracing returns False automatically. Please ' |
| | 'set on_trace manually if you are using trace.', UserWarning) |
| | return False |
| |
|
