Skip to content

hicaistar/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
format:  SavedModel
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
format:  SavedModel
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
format:  SavedModel
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

2 directories, 3 files

Tutorial

Please have a look at docs/tutorial.md

OCI Model Configuration Specification

Please have a look at docs/spec.md

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 87.1%
  • Makefile 11.1%
  • Shell 1.3%
  • Dockerfile 0.5%