File size: 251 Bytes
c985ba4
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
import os
from .default import DefaultEngineConfig


class EngineConfig(DefaultEngineConfig):
    def __init__(self, exp_name='default', model='AOTT'):
        super().__init__(exp_name, model)
        self.STAGE_NAME = 'YTB'

        self.init_dir()