diff --git a/main.py b/main.py index f05719d..28a5685 100644 --- a/main.py +++ b/main.py @@ -1,29 +1,38 @@ +# 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 -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 import photo_review -# 项目必须从此处启动,否则代码中的相对路径可能导致错误的发生 if __name__ == '__main__': logging.config.dictConfig(LOGGING_CONFIG) log = logging.getLogger() - - # 崩溃后的重试次数 - 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() + photo_review.main()