service_train_task.py 文件源码

def train(nn_id, wf_ver) :

    log_home = "/root"
    _celery_log_dir = make_celery_dir_by_datetime(log_home)
    celery_log_dir = make_and_exist_directory(_celery_log_dir)

    celery_log_file = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
    _filename = str(nn_id) +"_" +str(wf_ver) +"_" +  celery_log_file + ".log"

    celery_log_file = celery_log_dir + _filename

    logging.config.dictConfig(settings.LOGGING)
    logger = logging.getLogger()
    task_handler = FileHandler(celery_log_file)
    logger.addHandler(task_handler)
    logging.info("=============================================================")
    logging.info("[Train Task] Start Celery Job {0} {1}".format(nn_id, wf_ver))
    logging.info("=============================================================")
    result = WorkFlowTrainTask()._exec_train(nn_id, wf_ver)
    logging.info("=============================================================")
    logging.info("[Train Task] Done Celery Job {0} {1} : {2}".format(nn_id, wf_ver, result))
    logging.info("=============================================================")
    return result