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
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