fix: autotune pytest run forever (#132)

BaguaSys · Jul 20, 2021 · ef7399e · ef7399e
1 parent f8798bb
commit ef7399e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/service/test_autotune_service.py b/tests/service/test_autotune_service.py
@@ -6,6 +6,7 @@
 from bagua.service import AutotuneService, AutotuneClient
 from bagua.bagua_define import BaguaHyperparameter, get_tensor_declaration_bytes
 import socket
+import time
 import numpy as np
 import os
 
@@ -80,6 +81,7 @@ def run(self):
                 logging.info("train_iter={}".format(train_iter))
                 break
 
+            time.sleep(0.1)
             train_iter += 1
 
         return hp