Google Git
Sign in
apache / submarine / refs/heads/SUBMARINE-1393 / . / dev-support / examples / nn-pytorch
tree: 996612000d2ce53f21acb2e7a1a563e4a1325ee2 [path history] [tgz]
  1. build.sh
  2. Dockerfile
  3. model.py
  4. post.sh
  5. readme.md
dev-support/examples/nn-pytorch/readme.md

Save_model Example

Usage

This is an easy example of saving a pytorch linear model to model registry.

How to execute

  1. Build the docker image
./dev-support/examples/nn-pytorch/build.sh
  1. Submit a post request
./dev-support/examples/nn-pytorch/post.sh