Shortcuts

mmcv.utils.parrots_jit 源代码

# Copyright (c) OpenMMLab. All rights reserved.
import os

from mmengine.utils.dl_utils.parrots_wrapper import TORCH_VERSION

parrots_jit_option = os.getenv('PARROTS_JIT_OPTION')

if TORCH_VERSION == 'parrots' and parrots_jit_option == 'ON':
    from parrots.jit import pat as jit
else:

[文档] def jit(func=None, check_input=None, full_shape=True, derivate=False, coderize=False, optimize=False): def wrapper(func): def wrapper_inner(*args, **kargs): return func(*args, **kargs) return wrapper_inner if func is None: return wrapper else: return func
if TORCH_VERSION == 'parrots': from parrots.utils.tester import skip_no_elena else:
[文档] def skip_no_elena(func): def wrapper(*args, **kargs): return func(*args, **kargs) return wrapper
Read the Docs v: stable
Versions
latest
stable
2.x
v2.2.0
v2.1.0
v2.0.1
v2.0.0
1.x
v1.7.1
v1.7.0
v1.6.2
v1.6.1
v1.6.0
v1.5.3
v1.5.2_a
v1.5.1
v1.5.0
v1.4.8
v1.4.7
v1.4.6
v1.4.5
v1.4.4
v1.4.3
v1.4.2
v1.4.1
v1.4.0
v1.3.18
v1.3.17
v1.3.16
v1.3.15
v1.3.14
v1.3.13
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.