Sfoglia il codice sorgente

add warmup for python deploy

will-jl944 4 anni fa
parent
commit
7f055dc4eb
1 ha cambiato i file con 1 aggiunte e 2 eliminazioni
  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):