Exemplo n.º 1
0
def train_model():
    """

    :return:
    """
    if CFG.TRAIN.MULTI_GPU.ENABLE:
        LOG.info('Using multi gpu trainner ...')
        worker = multi_gpu_trainner.LaneNetTusimpleMultiTrainer()
    else:
        LOG.info('Using single gpu trainner ...')
        worker = single_gpu_trainner.LaneNetTusimpleTrainer()

    worker.train()
    return
Exemplo n.º 2
0
def train_model(): # 모델 학습 함수 정의
    """
    다중 GPU여부에 따라 구분하여 학습, 학습파일 경로는 tusimple_lanenet.yaml에서 설정
    :return:
    """
    if CFG.TRAIN.MULTI_GPU.ENABLE: # 다중 GPU 사용중일 경우
        LOG.info('Using multi gpu trainner ...')
        worker = multi_gpu_trainner.LaneNetTusimpleMultiTrainer(cfg=CFG) # 설정파일 이용하여 학습 객체 설정
    else: # 싱글 GPU 사용중일 경우
        LOG.info('Using single gpu trainner ...')
        worker = single_gpu_trainner.LaneNetTusimpleTrainer(cfg=CFG) # 설정파일 이용하여 학습 객체 설정

    worker.train() # 학습 시작
    print("학습 완료")
    return