Early Stopping Implementation

cmd/earlystopping/medianstop/v1beta1/Dockerfile

@@ -0,0 +1,22 @@
+FROM python:3.6
+
+ENV TARGET_DIR /opt/katib
+ENV EARLY_STOPPING_DIR cmd/earlystopping/medianstop/v1beta1
+
+RUN if [ "$(uname -m)" = "ppc64le" ] || [ "$(uname -m)" = "aarch64" ]; then \
+  apt-get -y update && \
+  apt-get -y install gfortran libopenblas-dev liblapack-dev && \
+  pip install cython; \
+  fi
+
+ADD ./pkg/ ${TARGET_DIR}/pkg/
+ADD ./${EARLY_STOPPING_DIR}/ ${TARGET_DIR}/${EARLY_STOPPING_DIR}/
+WORKDIR  ${TARGET_DIR}/${EARLY_STOPPING_DIR}
+RUN pip install --no-cache-dir -r requirements.txt
+
+RUN chgrp -R 0 ${TARGET_DIR} \
+  && chmod -R g+rwX ${TARGET_DIR}
+
+ENV PYTHONPATH ${TARGET_DIR}:${TARGET_DIR}/pkg/apis/manager/v1beta1/python
+
+ENTRYPOINT ["python", "main.py"]

cmd/earlystopping/medianstop/v1beta1/main.py

@@ -0,0 +1,31 @@
+import grpc
+import time
+import logging
+from pkg.apis.manager.v1beta1.python import api_pb2_grpc
+from pkg.earlystopping.v1beta1.medianstop.service import MedianStopService
+from concurrent import futures
+
+_ONE_DAY_IN_SECONDS = 60 * 60 * 24
+DEFAULT_PORT = "0.0.0.0:6788"
+
+logger = logging.getLogger()
+logging.basicConfig(level=logging.INFO)
+
+
+def serve():
+    server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
+    service = MedianStopService()
+    api_pb2_grpc.add_EarlyStoppingServicer_to_server(service, server)
+
+    server.add_insecure_port(DEFAULT_PORT)
+    logger.info("Start Median Stop service at address {}".format(DEFAULT_PORT))
+    server.start()
+    try:
+        while True:
+            time.sleep(_ONE_DAY_IN_SECONDS)
+    except KeyboardInterrupt:
+        server.stop(0)
+
+
+if __name__ == "__main__":
+    serve()

cmd/earlystopping/medianstop/v1beta1/requirements.txt

@@ -0,0 +1,4 @@
+grpcio==1.23.0
+protobuf==3.9.1
+googleapis-common-protos==1.6.0
+kubernetes==11.0.0