Пример #1
0
 def setUpClass(cls):
     onnx_model = os.path.join(cls.model_path, 'mnist.onnx')
     test_util.prepare_mnist_model(onnx_model)
     cmd = [cls.server_app_path, '--http_port', str(cls.server_port), '--model_path', onnx_model, '--log_level', cls.log_level]
     test_util.test_log('Launching server app: [{0}]'.format(' '.join(cmd)))
     cls.server_app_proc = subprocess.Popen(cmd)
     test_util.test_log('Server app PID: {0}'.format(cls.server_app_proc.pid))
     test_util.test_log('Sleep {0} second(s) to wait for server initialization'.format(cls.wait_server_ready_in_seconds))
     time.sleep(cls.wait_server_ready_in_seconds)
Пример #2
0
 def setUpClass(cls):
     onnx_model = os.path.join(cls.model_path, "model.onnx")
     test_util.prepare_mnist_model(onnx_model)
     cmd = [
         cls.server_app_path,
         "--grpc_port",
         str(cls.server_port),
         "--model_path",
         onnx_model,
         "--log_level",
         cls.log_level,
     ]
     test_util.test_log("Launching server app: [{0}]".format(" ".join(cmd)))
     cls.server_app_proc = subprocess.Popen(cmd)
     test_util.test_log("Server app PID: {0}".format(
         cls.server_app_proc.pid))
     test_util.test_log(
         "Sleep {0} second(s) to wait for server initialization".format(
             cls.wait_server_ready_in_seconds))
     time.sleep(cls.wait_server_ready_in_seconds)