You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

configuration_unite.py 420 B

123456789101112131415161718192021
  1. # Copyright (c) Alibaba, Inc. and its affiliates.
  2. """UniTE model configuration"""
  3. from enum import Enum
  4. from modelscope.utils import logger as logging
  5. from modelscope.utils.config import Config
  6. logger = logging.get_logger(__name__)
  7. class EvaluationMode(Enum):
  8. SRC = 'src'
  9. REF = 'ref'
  10. SRC_REF = 'src-ref'
  11. class UniTEConfig(Config):
  12. def __init__(self, **kwargs):
  13. super().__init__(**kwargs)