捕获到内存错误主动抛出中止程序，以释放显存

2024-11-28 10:27:44 +08:00
parent 9749577c5a
commit 8ea5420520
3 changed files with 6 additions and 1 deletions
--- a/photo_review/auto_photo_review.py
+++ b/photo_review/auto_photo_review.py
@@ -48,6 +48,9 @@ def ie_temp_image(ie, ocr, image):
            ie_result = []
        else:
            ie_result = ie({"doc": temp_file.name, "layout": layout})[0]
+    except MemoryError as e:
+        # 显存不足时应该抛出错误，让程序重启，同时释放显存
+        raise e
    except Exception as e:
        logging.error("信息抽取时出错", exc_info=e)
    finally: