File size: 513 Bytes
c985ba4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from .default import DefaultEngineConfig
class EngineConfig(DefaultEngineConfig):
def __init__(self, exp_name='default', model='AOTT'):
super().__init__(exp_name, model)
self.STAGE_NAME = 'PRE'
self.init_dir()
self.DATASETS = ['static']
self.DATA_DYNAMIC_MERGE_PROB = 1.0
self.TRAIN_LR = 4e-4
self.TRAIN_LR_MIN = 2e-5
self.TRAIN_WEIGHT_DECAY = 0.03
self.TRAIN_SEQ_TRAINING_START_RATIO = 1.0
self.TRAIN_AUX_LOSS_RATIO = 0.1
|