From 1718ff66b62c11b0348782bb0f3f2685fccdd03a Mon Sep 17 00:00:00 2001
From: cuicheng01 <cuicheng_smile@163.com>
Date: Thu, 25 Nov 2021 12:36:53 +0000
Subject: [PATCH] update predictor.py

---
 deploy/utils/predictor.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/deploy/utils/predictor.py b/deploy/utils/predictor.py
index 11f15307..1c49c9dd 100644
--- a/deploy/utils/predictor.py
+++ b/deploy/utils/predictor.py
@@ -60,6 +60,7 @@ class Predictor(object):
                 precision_mode=Config.Precision.Half
                 if args.use_fp16 else Config.Precision.Float32,
                 max_batch_size=args.batch_size,
+                workspace_size=1 << 30,
                 min_subgraph_size=30)
 
         config.enable_memory_optim()
-- 
GitLab