Skip to content
Snippets Groups Projects
Unverified Commit 56cd6a38 authored by leaves-zwx's avatar leaves-zwx Committed by GitHub
Browse files

Implementation of SavedModel and InferenceSession (#4066)


* save model and load model demo

* fix

* tensor.proto and copy signature

* pass test

* add load_saved_model function for InferenceSession

* wait_for_all_jobs_finished

* test_alexnet_save_and_load

* support change batch_size

* support batch axis

* add ci test

* revert job_build_and_infer_ctx api

* simplify test script

* following update

* improve search function of InferenceSession

* fix break update

* add cv2 to dev-requirements

* Update Dockerfile

* rm 3.5

* fix

* quick workaround

* speed up bazel

* port changes

* revert workaround

* rm batch_axis in JobInputDef and JobOutputDef

* rm export for ImageNetRecordDataset

* refine Complete api for GraphBuilder and SignatureBuilder

* refine check op is moirrored

* fix by review comment ci test

* InferenceSession is not responsible to destroy env

Co-authored-by: default avatarShenghang Tsai <jackalcooper@gmail.com>
Co-authored-by: default avatarTsai <caishenghang@oneflow.org>
Co-authored-by: default avataroneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
parent 5a4dd2a8
No related branches found
No related tags found
No related merge requests found
Showing
with 560 additions and 10 deletions
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment