冲突排查

This commit is contained in:
2024-06-21 15:35:10 +08:00
parent 2005b038b9
commit 61d7bc1457

56
main.py
View File

@@ -1,29 +1,33 @@
import logging.config # import logging.config
import os # import os
import traceback # import traceback
#
from config.photo_review import RETRY_TIME, SEND_ERROR_EMAIL, CUDA_VISIBLE_DEVICES # from config.photo_review import RETRY_TIME, SEND_ERROR_EMAIL, CUDA_VISIBLE_DEVICES
#
os.environ["CUDA_VISIBLE_DEVICES"] = CUDA_VISIBLE_DEVICES # os.environ["CUDA_VISIBLE_DEVICES"] = CUDA_VISIBLE_DEVICES
#
import paddle # import paddle
from auto_email.error_email import send_an_error_email # from auto_email.error_email import send_an_error_email
from config.log import LOGGING_CONFIG # 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()
from photo_review.photo_review import main from photo_review.photo_review import main
# 项目必须从此处启动,否则代码中的相对路径可能导致错误的发生
if __name__ == '__main__': if __name__ == '__main__':
logging.config.dictConfig(LOGGING_CONFIG) main()
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()