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