Преглед на файлове

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