ソースを参照

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):