diff --git a/main.py b/main.py index 2efff18..f05719d 100644 --- a/main.py +++ b/main.py @@ -1,52 +1,29 @@ -# import logging.config -# import os -# import traceback -# -# from config.photo_review import RETRY_TIME, SEND_ERROR_EMAIL, CUDA_VISIBLE_DEVICES -# -# os.environ["CUDA_VISIBLE_DEVICES"] = CUDA_VISIBLE_DEVICES -# -# import paddle -# from auto_email.error_email import send_an_error_email -# from config.log import LOGGING_CONFIG -# from photo_review.photo_review import main -# -# # 项目必须从此处启动,否则代码中的相对路径可能导致错误的发生 -# if __name__ == '__main__': -# logging.config.dictConfig(LOGGING_CONFIG) -# log = logging.getLogger() -# -# # 崩溃后的重试次数 -# for _ in range(RETRY_TIME + 1): -# try: -# log.info("【照片审核关键信息抽取】开始") -# main() -# except Exception as e: -# log.error(traceback.format_exc()) -# if SEND_ERROR_EMAIL: -# send_an_error_email(program_name='照片审核关键信息抽取脚本', error_name=repr(e), error_detail=traceback.format_exc()) -# # 释放显存 -# paddle.device.cuda.empty_cache() import logging.config +import os import traceback -import paddle +from config.photo_review import RETRY_TIME, SEND_ERROR_EMAIL, CUDA_VISIBLE_DEVICES +os.environ["CUDA_VISIBLE_DEVICES"] = CUDA_VISIBLE_DEVICES + +import paddle from auto_email.error_email import send_an_error_email from config.log import LOGGING_CONFIG -from config.photo_review import SEND_ERROR_EMAIL from photo_review import photo_review +# 项目必须从此处启动,否则代码中的相对路径可能导致错误的发生 if __name__ == '__main__': logging.config.dictConfig(LOGGING_CONFIG) log = logging.getLogger() - try: - log.info("【照片审核关键信息抽取】开始") - photo_review.main() - except Exception as e: - log.error(traceback.format_exc()) - if SEND_ERROR_EMAIL: - send_an_error_email(program_name='照片审核关键信息抽取脚本', error_name=repr(e), error_detail=traceback.format_exc()) - # 释放显存 - paddle.device.cuda.empty_cache() + # 崩溃后的重试次数 + for _ in range(RETRY_TIME + 1): + try: + # log.info("【照片审核关键信息抽取】开始") + photo_review.main() + except Exception as e: + log.error(traceback.format_exc()) + if SEND_ERROR_EMAIL: + send_an_error_email(program_name='照片审核关键信息抽取脚本', error_name=repr(e), error_detail=traceback.format_exc()) + # 释放显存 + paddle.device.cuda.empty_cache()