暂时取消主动释放显存

2024-06-26 13:47:31 +08:00
parent 403858a287
commit 3645e13f00
2 changed files with 3 additions and 19 deletions
--- a/photo_review/photo_review.py
+++ b/photo_review/photo_review.py
@@ -173,9 +173,9 @@ def information_extraction(ie, phrecs):

            result = merge_result(result, ie_result)

-    # 完成一次任务释放显存
-    paddle.device.cuda.empty_cache()
-    logging.info("显存释放完成")
+    # # 完成一次任务释放显存
+    # paddle.device.cuda.empty_cache()
+    # logging.info("显存释放完成")
    return result