Spaces:
Sleeping
Sleeping
# Copyright (c) OpenMMLab. All rights reserved. | |
from .decorators import auto_fp16, force_fp32 | |
from .hooks import Fp16OptimizerHook, wrap_fp16_model | |
from .utils import cast_tensor_type | |
__all__ = [ | |
'auto_fp16', 'force_fp32', 'Fp16OptimizerHook', 'wrap_fp16_model', | |
'cast_tensor_type' | |
] | |