|
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 |
|
|
