فهرست منبع

add warmup for python deploy

will-jl944 4 سال پیش
والد
کامیت
7f055dc4eb
1فایلهای تغییر یافته به همراه1 افزوده شده و 2 حذف شده
  1. 1 2
      paddlex/deploy.py

+ 1 - 2
paddlex/deploy.py

@@ -263,8 +263,7 @@ class Predictor(object):
             images = img_file
 
         for step in range(warmup_iters):
-            self._run(
-                images=images, topk=topk, transforms=transforms, verbose=False)
+            self._run(images=images, topk=topk, transforms=transforms)
         self.timer.reset()
 
         for step in range(repeats):