__init__.py 699 B

12345678910111213141516171819202122232425
  1. # Copyright (c) Alibaba, Inc. and its affiliates.
  2. from typing import TYPE_CHECKING
  3. from modelscope.utils.import_utils import LazyImportModule
  4. if TYPE_CHECKING:
  5. from .builder import LR_SCHEDULER, build_lr_scheduler
  6. from .warmup import BaseWarmup, ConstantWarmup, ExponentialWarmup, LinearWarmup
  7. else:
  8. _import_structure = {
  9. 'builder': ['LR_SCHEDULER', 'build_lr_scheduler'],
  10. 'warmup':
  11. ['BaseWarmup', 'ConstantWarmup', 'ExponentialWarmup', 'LinearWarmup']
  12. }
  13. import sys
  14. sys.modules[__name__] = LazyImportModule(
  15. __name__,
  16. globals()['__file__'],
  17. _import_structure,
  18. module_spec=__spec__,
  19. extra_objects={},
  20. )