aot/configs/models/default_deaot.py

from .default import DefaultModelConfig as BaseConfig


class DefaultModelConfig(BaseConfig):
    def __init__(self):
        super().__init__()
        self.MODEL_NAME = 'DeAOTDefault'

        self.MODEL_VOS = 'deaot'
        self.MODEL_ENGINE = 'deaotengine'

        self.MODEL_DECODER_INTERMEDIATE_LSTT = False

        self.MODEL_SELF_HEADS = 1
        self.MODEL_ATT_HEADS = 1

        self.TRAIN_AUG_TYPE = 'v2'