expose gunicorn logging (#219)

Author: mseth10

docker/build_artifacts/sagemaker/serve.py

@@ -51,6 +51,7 @@ def __init__(self):
         self._sagemaker_port_range = os.environ.get("SAGEMAKER_SAFE_PORT_RANGE", None)
         self._gunicorn_workers = os.environ.get("SAGEMAKER_GUNICORN_WORKERS", 1)
         self._gunicorn_threads = os.environ.get("SAGEMAKER_GUNICORN_THREADS", 1)
+        self._gunicorn_loglevel = os.environ.get("SAGEMAKER_GUNICORN_LOGLEVEL", "info")
         self._tfs_config_path = "/sagemaker/model-config.cfg"
         self._tfs_batching_config_path = "/sagemaker/batching-config.cfg"
 
@@ -204,7 +205,7 @@ def _setup_gunicorn(self):
 
         gunicorn_command = (
             "gunicorn -b unix:/tmp/gunicorn.sock -k {} --chdir /sagemaker "
-            "--workers {} --threads {} "
+            "--workers {} --threads {} --log-level {} "
             "{}{} -e TFS_GRPC_PORTS={} -e TFS_REST_PORTS={} "
             "-e SAGEMAKER_MULTI_MODEL={} -e SAGEMAKER_SAFE_PORT_RANGE={} "
             "-e SAGEMAKER_TFS_WAIT_TIME_SECONDS={} "
@@ -213,6 +214,7 @@ def _setup_gunicorn(self):
             self._gunicorn_worker_class,
             self._gunicorn_workers,
             self._gunicorn_threads,
+            self._gunicorn_loglevel,
             python_path_option,
             ",".join(python_path_content),
             self._tfs_grpc_concat_ports,