diff --git a/main.py b/main.py index edaf0d6..2efff18 100644 --- a/main.py +++ b/main.py @@ -28,15 +28,25 @@ # # 释放显存 # paddle.device.cuda.empty_cache() import logging.config +import traceback +import paddle + +from auto_email.error_email import send_an_error_email from config.log import LOGGING_CONFIG -from config.photo_review import RETRY_TIME +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() - 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()