27 lines
881 B
Python
27 lines
881 B
Python
import logging.config
|
|
import os
|
|
import traceback
|
|
|
|
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
|
|
|
|
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(2):
|
|
try:
|
|
log.info("【照片审核关键信息抽取】开始")
|
|
main()
|
|
except Exception as e:
|
|
log.error(traceback.format_exc())
|
|
send_an_error_email(program_name='照片审核关键信息抽取脚本', error_name=repr(e), error_detail=traceback.format_exc())
|
|
# 释放显存
|
|
paddle.device.cuda.empty_cache()
|