aikenml's picture
Upload folder using huggingface_hub
c985ba4
raw
history blame
289 Bytes
from .default_deaot import DefaultModelConfig
class ModelConfig(DefaultModelConfig):
def __init__(self):
super().__init__()
self.MODEL_NAME = 'DeAOTL'
self.MODEL_LSTT_NUM = 3
self.TRAIN_LONG_TERM_MEM_GAP = 2
self.TEST_LONG_TERM_MEM_GAP = 5