예제 #1
0
def register_results(cls, name=None):
    return register(cls, RESULTS, name, 'results')
예제 #2
0
def register_arc_policy(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_SEQ2SEQ_ARC_POLICY, name, 'decoder')
예제 #3
0
def register_reporting(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_REPORTING, name, 'reporting')
예제 #4
0
def register_vectorizer(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_VECTORIZERS, name, 'vectorizer')
예제 #5
0
def register_decoder(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_SEQ2SEQ_DECODERS, name, 'decoder')
예제 #6
0
def register_reporting(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_REPORTING, name, 'reporting')
예제 #7
0
def register_vectorizer(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_VECTORIZERS, name, 'vectorizer')
예제 #8
0
def register_preprocessor(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_PREPROCESSORS, name, 'preprocessor')
예제 #9
0
def register_sampler(cls, name=None):
    if cls.name is None:
        cls.name = name if name is not None else cls.__name__
    return register(cls, SAMPLERS, cls.name, "sampler")
예제 #10
0
파일: model.py 프로젝트: dpressel/baseline
def register_arc_policy(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_SEQ2SEQ_ARC_POLICY, name, 'decoder')
예제 #11
0
def register_progress(cls, name=None):
    return register(cls, BASELINE_PROGRESS, name, 'progress')
예제 #12
0
파일: model.py 프로젝트: dpressel/baseline
def register_decoder(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_SEQ2SEQ_DECODERS, name, 'decoder')
예제 #13
0
def register_backend(cls, name=None):
    return register(cls, BACKENDS, name, "backend")
예제 #14
0
def register_backend(cls, name=None):
    return register(cls, BACKENDS, name, "backend")
예제 #15
0
def register_lr_scheduler(cls, name=None):
    return register(cls, BASELINE_LR_SCHEDULERS, name, 'lr_scheduler')
예제 #16
0
def register_frontend(cls, name=None):
    return register(cls, FRONTENDS, name, "frontend")
예제 #17
0
파일: train.py 프로젝트: dpressel/baseline
def register_lr_scheduler(cls, name=None):
    return register(cls, BASELINE_LR_SCHEDULERS, name, 'lr_scheduler')
예제 #18
0
def register_preprocessor(cls, name=None):
    """Register a function as a plug-in"""
    return register(cls, BASELINE_PREPROCESSORS, name, 'preprocessor')