Skip to content

Commit 15963b0

Browse files
fix: release memory after predict (#10688)
* fix: release memory after predict * fix: release memory
1 parent c13a50c commit 15963b0

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tools/infer/predict_rec.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -624,6 +624,7 @@ def __call__(self, img_list):
624624
preds = outputs
625625
else:
626626
preds = outputs[0]
627+
self.predictor.try_shrink_memory()
627628
rec_result = self.postprocess_op(preds)
628629
for rno in range(len(rec_result)):
629630
rec_res[indices[beg_img_no + rno]] = rec_result[rno]

0 commit comments

Comments
 (0)