diff --git a/tests/service/test_autotune_service.py b/tests/service/test_autotune_service.py index e7fa31195..9d1ff7224 100644 --- 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