Skip to content

Commit

Permalink
*: Update name (#64)
Browse files Browse the repository at this point in the history
Signed-off-by: Ce Gao <[email protected]>
  • Loading branch information
gaocegege authored Apr 20, 2018
1 parent dd6c87e commit 895aaab
Show file tree
Hide file tree
Showing 53 changed files with 124 additions and 124 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# Katib

[![Go Report Card](https://goreportcard.com/badge/github.com/kubeflow/hp-tuning)](https://goreportcard.com/report/github.com/kubeflow/hp-tuning)
[![Go Report Card](https://goreportcard.com/badge/github.com/kubeflow/katib)](https://goreportcard.com/report/github.com/kubeflow/katib)

Hyperparameter Tuning on Kubernetes.
This project is inspired by [Google vizier](https://static.googleusercontent.com/media/research.google.com/ja//pubs/archive/bcb15507f4b52991a0783013df4222240e942381.pdf). Katib is a scalable and flexible hyperparameter tuning framework and is tightly integrated with kubernetes. Also it does not depend on a specific Deep Learning framework e.g. TensorFlow, MXNet, and PyTorch).
Expand Down
6 changes: 3 additions & 3 deletions cli/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
FROM golang:alpine AS build-env
# The GOPATH in the image is /go.
ADD . /go/src/github.com/kubeflow/hp-tuning
WORKDIR /go/src/github.com/kubeflow/hp-tuning/cli
ADD . /go/src/github.com/kubeflow/katib
WORKDIR /go/src/github.com/kubeflow/katib/cli
RUN go build -o katib-cli

FROM alpine:3.7
WORKDIR /app
COPY --from=build-env /go/src/github.com/kubeflow/hp-tuning/cli/katib-cli /app/
COPY --from=build-env /go/src/github.com/kubeflow/katib/cli/katib-cli /app/
2 changes: 1 addition & 1 deletion cli/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ import (

"google.golang.org/grpc"

pb "github.com/kubeflow/hp-tuning/api"
pb "github.com/kubeflow/katib/api"
)

var server = flag.String("s", "127.0.0.1:6789", "server address")
Expand Down
2 changes: 1 addition & 1 deletion db/interface.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ import (
"strings"
"time"

api "github.com/kubeflow/hp-tuning/api"
api "github.com/kubeflow/katib/api"

_ "github.com/go-sql-driver/mysql"
)
Expand Down
2 changes: 1 addition & 1 deletion db/interface_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ import (
"github.com/golang/protobuf/jsonpb"
"gopkg.in/DATA-DOG/go-sqlmock.v1"

api "github.com/kubeflow/hp-tuning/api"
api "github.com/kubeflow/katib/api"
)

var db_interface VizierDBInterface
Expand Down
2 changes: 1 addition & 1 deletion db/test/test.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ package main

import (
"fmt"
"github.com/kubeflow/hp-tuning/db"
"github.com/kubeflow/katib/db"
"os"
)

Expand Down
8 changes: 4 additions & 4 deletions dlk/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
FROM golang:alpine AS build-env
# The GOPATH in the image is /go.
ADD . /go/src/github.com/kubeflow/hp-tuning
WORKDIR /go/src/github.com/kubeflow/hp-tuning/dlk/dlkmanager
RUN go build -o dlkmanager
ADD . /go/src/github.com/kubeflow/katib
WORKDIR /go/src/github.com/kubeflow/katib/dlk/dlkmanager
RUN go build -o dlkmanager

FROM alpine:3.7
WORKDIR /app
COPY --from=build-env /go/src/github.com/kubeflow/hp-tuning/dlk/dlkmanager/dlkmanager /app/
COPY --from=build-env /go/src/github.com/kubeflow/katib/dlk/dlkmanager/dlkmanager /app/
ENTRYPOINT ["./dlkmanager"]
2 changes: 1 addition & 1 deletion dlk/dlkctl/del-lt.go
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ import (
"net/http"
"os"

"github.com/kubeflow/hp-tuning/dlk/dlkctl/utils"
"github.com/kubeflow/katib/dlk/dlkctl/utils"
"github.com/spf13/cobra"
)

Expand Down
4 changes: 2 additions & 2 deletions dlk/dlkctl/get-lt.go
Original file line number Diff line number Diff line change
Expand Up @@ -22,9 +22,9 @@ import (
"os"
"sort"

"github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore"
"github.com/kubeflow/katib/dlk/dlkmanager/datastore"

"github.com/kubeflow/hp-tuning/dlk/dlkctl/utils"
"github.com/kubeflow/katib/dlk/dlkctl/utils"
"github.com/spf13/cobra"
)

Expand Down
4 changes: 2 additions & 2 deletions dlk/dlkctl/get-lts.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,9 @@ import (
"net/http"
"os"

"github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore"
"github.com/kubeflow/katib/dlk/dlkmanager/datastore"

"github.com/kubeflow/hp-tuning/dlk/dlkctl/utils"
"github.com/kubeflow/katib/dlk/dlkctl/utils"
"github.com/spf13/cobra"
)

Expand Down
4 changes: 2 additions & 2 deletions dlk/dlkctl/logs.go
Original file line number Diff line number Diff line change
Expand Up @@ -23,8 +23,8 @@ import (
"os"
"time"

"github.com/kubeflow/hp-tuning/dlk/dlkctl/utils"
"github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore"
"github.com/kubeflow/katib/dlk/dlkctl/utils"
"github.com/kubeflow/katib/dlk/dlkmanager/datastore"
"github.com/spf13/cobra"
)

Expand Down
4 changes: 2 additions & 2 deletions dlk/dlkctl/run.go
Original file line number Diff line number Diff line change
Expand Up @@ -19,8 +19,8 @@ import (
"strings"
"time"

"github.com/kubeflow/hp-tuning/dlk/dlkctl/utils"
"github.com/kubeflow/hp-tuning/dlk/dlkmanager/api"
"github.com/kubeflow/katib/dlk/dlkctl/utils"
"github.com/kubeflow/katib/dlk/dlkmanager/api"

"github.com/spf13/cobra"
)
Expand Down
2 changes: 1 addition & 1 deletion dlk/dlkmanager/delete.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ import (
"fmt"
"net/http"

"github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore"
"github.com/kubeflow/katib/dlk/dlkmanager/datastore"
"github.com/labstack/echo"
lgr "github.com/sirupsen/logrus"
)
Expand Down
2 changes: 1 addition & 1 deletion dlk/dlkmanager/dlkmanager.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ import (
"fmt"
"os"

"github.com/kubeflow/hp-tuning/dlk/dlkmanager/configs"
"github.com/kubeflow/katib/dlk/dlkmanager/configs"
"github.com/labstack/echo" // Web server framework for REST API
lgr "github.com/sirupsen/logrus" // logging framework
"github.com/spf13/cobra"
Expand Down
2 changes: 1 addition & 1 deletion dlk/dlkmanager/get.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ import (
"net/http"
"time"

"github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore"
"github.com/kubeflow/katib/dlk/dlkmanager/datastore"
"github.com/labstack/echo"
lgr "github.com/sirupsen/logrus"
)
Expand Down
4 changes: 2 additions & 2 deletions dlk/dlkmanager/learningTask.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,8 @@ import (
"sync"
"time"

"github.com/kubeflow/hp-tuning/dlk/dlkmanager/api"
"github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore"
"github.com/kubeflow/katib/dlk/dlkmanager/api"
"github.com/kubeflow/katib/dlk/dlkmanager/datastore"

lgr "github.com/sirupsen/logrus"
batchv1 "k8s.io/api/batch/v1"
Expand Down
6 changes: 3 additions & 3 deletions dlk/dlkmanager/post.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,10 +6,10 @@ import (
"path/filepath"
"time"

"github.com/kubeflow/hp-tuning/dlk/dlkmanager/configs"
"github.com/kubeflow/katib/dlk/dlkmanager/configs"

"github.com/kubeflow/hp-tuning/dlk/dlkmanager/api"
"github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore"
"github.com/kubeflow/katib/dlk/dlkmanager/api"
"github.com/kubeflow/katib/dlk/dlkmanager/datastore"
corev1 "k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/client-go/kubernetes"
Expand Down
2 changes: 1 addition & 1 deletion docs/getting-start.md
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
First, Copy CLI tool.

```bash
$ curl -Lo katib-cli https://github.com/kubeflow/hp-tuning/releases/download/v0.1.0-alpha/katib-cli-linux-amd64 && chmod +x katib-cli && sudo mv katib-cli /usr/local/bin/
$ curl -Lo katib-cli https://github.com/kubeflow/katib/releases/download/v0.1.0-alpha/katib-cli-linux-amd64 && chmod +x katib-cli && sudo mv katib-cli /usr/local/bin/
```

The cli tool will be put `/usr/local/bin/` directory.
Expand Down
4 changes: 2 additions & 2 deletions earlystopping/medianstopping.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,8 @@ package earlystopping
import (
"context"
"errors"
"github.com/kubeflow/hp-tuning/api"
vdb "github.com/kubeflow/hp-tuning/db"
"github.com/kubeflow/katib/api"
vdb "github.com/kubeflow/katib/db"
"log"
"sort"
"strconv"
Expand Down
6 changes: 3 additions & 3 deletions earlystopping/medianstopping/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
FROM golang:alpine AS build-env
# The GOPATH in the image is /go.
ADD . /go/src/github.com/kubeflow/hp-tuning
WORKDIR /go/src/github.com/kubeflow/hp-tuning/earlystopping/medianstopping
ADD . /go/src/github.com/kubeflow/katib
WORKDIR /go/src/github.com/kubeflow/katib/earlystopping/medianstopping
RUN go build -o medianstopping

FROM alpine:3.7
WORKDIR /app
COPY --from=build-env /go/src/github.com/kubeflow/hp-tuning/earlystopping/medianstopping /app/
COPY --from=build-env /go/src/github.com/kubeflow/katib/earlystopping/medianstopping /app/
ENTRYPOINT ["./medianstopping"]
4 changes: 2 additions & 2 deletions earlystopping/medianstopping/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,8 @@ import (
"google.golang.org/grpc"
"google.golang.org/grpc/reflection"

pb "github.com/kubeflow/hp-tuning/api"
"github.com/kubeflow/hp-tuning/earlystopping"
pb "github.com/kubeflow/katib/api"
"github.com/kubeflow/katib/earlystopping"
)

func main() {
Expand Down
2 changes: 1 addition & 1 deletion earlystopping/types.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ package earlystopping
import (
"context"

"github.com/kubeflow/hp-tuning/api"
"github.com/kubeflow/katib/api"
)

const (
Expand Down
8 changes: 4 additions & 4 deletions manager/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
FROM golang:alpine AS build-env
# The GOPATH in the image is /go.
ADD . /go/src/github.com/kubeflow/hp-tuning
WORKDIR /go/src/github.com/kubeflow/hp-tuning/manager
ADD . /go/src/github.com/kubeflow/katib
WORKDIR /go/src/github.com/kubeflow/katib/manager
RUN go build -o vizier-manager

FROM alpine:3.7
WORKDIR /app
COPY --from=build-env /go/src/github.com/kubeflow/hp-tuning/manager/vizier-manager /app/
COPY --from=build-env /go/src/github.com/kubeflow/hp-tuning/manager/visualise /
COPY --from=build-env /go/src/github.com/kubeflow/katib/manager/vizier-manager /app/
COPY --from=build-env /go/src/github.com/kubeflow/katib/manager/visualise /
ENTRYPOINT ["./vizier-manager"]
CMD ["-w", "dlk"]
16 changes: 8 additions & 8 deletions manager/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,14 +9,14 @@ import (
"strconv"
"time"

pb "github.com/kubeflow/hp-tuning/api"
kdb "github.com/kubeflow/hp-tuning/db"
"github.com/kubeflow/hp-tuning/manager/modelstore"
tbif "github.com/kubeflow/hp-tuning/manager/visualise/tensorboard"
"github.com/kubeflow/hp-tuning/manager/worker_interface"
dlkwif "github.com/kubeflow/hp-tuning/manager/worker_interface/dlk"
k8swif "github.com/kubeflow/hp-tuning/manager/worker_interface/kubernetes"
nvdwif "github.com/kubeflow/hp-tuning/manager/worker_interface/nvdocker"
pb "github.com/kubeflow/katib/api"
kdb "github.com/kubeflow/katib/db"
"github.com/kubeflow/katib/manager/modelstore"
tbif "github.com/kubeflow/katib/manager/visualise/tensorboard"
"github.com/kubeflow/katib/manager/worker_interface"
dlkwif "github.com/kubeflow/katib/manager/worker_interface/dlk"
k8swif "github.com/kubeflow/katib/manager/worker_interface/kubernetes"
nvdwif "github.com/kubeflow/katib/manager/worker_interface/nvdocker"

"google.golang.org/grpc"
"google.golang.org/grpc/reflection"
Expand Down
8 changes: 4 additions & 4 deletions manager/main_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,10 +6,10 @@ import (

"github.com/golang/mock/gomock"

api "github.com/kubeflow/hp-tuning/api"
mockdb "github.com/kubeflow/hp-tuning/mock/db"
mockmodelstore "github.com/kubeflow/hp-tuning/mock/modelstore"
mockworker "github.com/kubeflow/hp-tuning/mock/worker"
api "github.com/kubeflow/katib/api"
mockdb "github.com/kubeflow/katib/mock/db"
mockmodelstore "github.com/kubeflow/katib/mock/modelstore"
mockworker "github.com/kubeflow/katib/mock/worker"
)

func TestCreateStudy(t *testing.T) {
Expand Down
4 changes: 2 additions & 2 deletions manager/modelstore/modeldb.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,8 @@ import (
"context"
"fmt"
"git.apache.org/thrift.git/lib/go/thrift"
"github.com/kubeflow/hp-tuning/api"
"github.com/kubeflow/hp-tuning/manager/modelstore/modeldb"
"github.com/kubeflow/katib/api"
"github.com/kubeflow/katib/manager/modelstore/modeldb"
"log"
"net"
"strconv"
Expand Down

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion manager/modelstore/modelstore.go
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
package modelstore

import (
"github.com/kubeflow/hp-tuning/api"
"github.com/kubeflow/katib/api"
)

type ModelStore interface {
Expand Down
2 changes: 1 addition & 1 deletion manager/visualise/tensorboard/tensorboardIF.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ package tensorboard

import (
"bytes"
"github.com/kubeflow/hp-tuning/api"
"github.com/kubeflow/katib/api"
"io/ioutil"
apiv1 "k8s.io/api/core/v1"
exbeatav1 "k8s.io/api/extensions/v1beta1"
Expand Down
10 changes: 5 additions & 5 deletions manager/worker_interface/dlk/dlk.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,11 +4,11 @@ import (
"bytes"
"encoding/json"
"fmt"
"github.com/kubeflow/hp-tuning/api"
"github.com/kubeflow/hp-tuning/db"
dlkapi "github.com/kubeflow/hp-tuning/dlk/dlkmanager/api"
"github.com/kubeflow/hp-tuning/dlk/dlkmanager/datastore"
wIF "github.com/kubeflow/hp-tuning/manager/worker_interface"
"github.com/kubeflow/katib/api"
"github.com/kubeflow/katib/db"
dlkapi "github.com/kubeflow/katib/dlk/dlkmanager/api"
"github.com/kubeflow/katib/dlk/dlkmanager/datastore"
wIF "github.com/kubeflow/katib/manager/worker_interface"
"io/ioutil"
"log"
"net/http"
Expand Down
8 changes: 4 additions & 4 deletions manager/worker_interface/kubernetes/kubernetes.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,10 @@ import (
"bytes"
"errors"
"fmt"
"github.com/kubeflow/hp-tuning/api"
"github.com/kubeflow/hp-tuning/db"
"github.com/kubeflow/hp-tuning/earlystopping"
wIF "github.com/kubeflow/hp-tuning/manager/worker_interface"
"github.com/kubeflow/katib/api"
"github.com/kubeflow/katib/db"
"github.com/kubeflow/katib/earlystopping"
wIF "github.com/kubeflow/katib/manager/worker_interface"
"io/ioutil"
batchv1 "k8s.io/api/batch/v1"
apiv1 "k8s.io/api/core/v1"
Expand Down
6 changes: 3 additions & 3 deletions manager/worker_interface/nvdocker/nvidia-docker.go
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@ import (
"github.com/docker/docker/api/types"
"github.com/docker/docker/api/types/container"
dclient "github.com/docker/docker/client"
"github.com/kubeflow/hp-tuning/api"
"github.com/kubeflow/hp-tuning/db"
wIF "github.com/kubeflow/hp-tuning/manager/worker_interface"
"github.com/kubeflow/katib/api"
"github.com/kubeflow/katib/db"
wIF "github.com/kubeflow/katib/manager/worker_interface"
"io"
"io/ioutil"
"log"
Expand Down
4 changes: 2 additions & 2 deletions manager/worker_interface/worker_interface.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,8 @@ package worker_interface
import (
"errors"
"fmt"
"github.com/kubeflow/hp-tuning/api"
"github.com/kubeflow/hp-tuning/db"
"github.com/kubeflow/katib/api"
"github.com/kubeflow/katib/db"
"time"
)

Expand Down
4 changes: 2 additions & 2 deletions mock/api/manager.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading

0 comments on commit 895aaab

Please sign in to comment.