Skip to content
/ ormb Public
forked from kleveross/ormb

OCI Registry for Model Bundle

Notifications You must be signed in to change notification settings

tskdsb/ormb

 
 

Repository files navigation

Table of Contents generated with DocToc

ormb

Managing Machine Learning/Deep Learning Models like Docker Images.

ormb helps you manage your Machine Learning/Deep Learning models. It makes your models easy to create, version, share and publish.

ormb can publish models by reading the information from a ormbfile.yaml. A ormbfile.yaml is a text document that contains all the information about the model.

Installation

go install github.com/caicloud/ormb

Usage

Save the model

$ ormb save ./resnet_v2_fp32_savedmodel_NCHW caicloud/resnetv2:v1
ref:     caicloud/resnetv2:v1
digest:  f51973c855608ab06d8f5e4333925a635f87f01ff992ffc5f9988f26d1da24e9
size:    90.6 MiB
v1: saved

Push the model to a remote registry

$ ormb push caicloud/resnetv2:v1
The push refers to repository [caicloud/resnetv2]
ref:     caicloud/resnetv2:v1
digest:  f51973c855608ab06d8f5e4333925a635f87f01ff992ffc5f9988f26d1da24e9
size:    90.6 MiB
v1: pushed to remote (1 layer, 90.6 MiB total)

Pull the model from a remote registry

$ ormb pull caicloud/resnetv2:v1 
v1: Pulling from caicloud/resnetv2
ref:     caicloud/resnetv2:v1
digest:  f51973c855608ab06d8f5e4333925a635f87f01ff992ffc5f9988f26d1da24e9
size:    90.6 MiB
Status: Downloaded newer model for caicloud/resnetv2:v1

Export the model to the current directory

$ ormb export caicloud/resnetv2:v1
ref:     localhost:5000/caicloud/resnetv2:v1
digest:  f51973c855608ab06d8f5e4333925a635f87f01ff992ffc5f9988f26d1da24e9
size:    90.6 MiB

$ tree ./resnet_v2_fp32_savedmodel_NCHW
resnet_v2_fp32_savedmodel_NCHW
├── 1538687196
│   ├── saved_model.pb
│   └── variables
│       ├── variables.data-00000-of-00001
│       └── variables.index
└── ormbfile.yaml

2 directories, 4 files

Community

ormb project is part of Clever, a Cloud Native Machine Learning platform. We are going to open source a community edition soon.

The Clever slack workspace is caicloud-clever.slack.com. To join, click this invitation to our Slack workspace.

About

OCI Registry for Model Bundle

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Go 86.1%
  • Makefile 12.0%
  • Shell 1.4%
  • Dockerfile 0.5%