Skip to content

Commit

Permalink
修改wideAndDeep和pytorch模型
Browse files Browse the repository at this point in the history
  • Loading branch information
fengyu10 committed Jun 12, 2020
1 parent de1b21e commit f4734be
Show file tree
Hide file tree
Showing 46 changed files with 162 additions and 5,842 deletions.
7 changes: 4 additions & 3 deletions DLPredictOnline/demo/README.MD
Original file line number Diff line number Diff line change
Expand Up @@ -9,8 +9,9 @@ Tensorflow模型部署步骤如下:
2. 运行容器,启动tensorflow-serving,绑定到宿主机8501端口上,命令:
docker run -t --rm -p 8501:8501 -p 8500:8500 -v "$TESTDATA:/models/textcnn-69" -e MODEL_NAME=textcnn-69 tensorflow/serving:1.14.0 &
3. -e MODEL_NAME=textcnn-69 表示模型名称,其中69表示此次测试的任务id,由用户自己指定
4. -p 8501:8501 http端口 表示将宿主机端口8051绑定到容器8051端口上,这样可以通过物理机8051端口可以直接访问容器中服务
5. -p 8500:8500 grpc端口
4. -v /models/textcnn-69 容器内部目录必须符合/models/$MODEL_NAME 规范
5. -p 8501:8501 http端口 表示将宿主机端口8051绑定到容器8051端口上,这样可以通过物理机8051端口可以直接访问容器中服务
6. -p 8500:8500 grpc端口

模型启动日志如下所示:
![模型启动日志](tensorflow-serving start log.png)
Expand Down Expand Up @@ -133,7 +134,7 @@ key=__label__thankyou, value=1.5074686E-5

PyTorch模型部署步骤如下:

1. 下载[待部署的PyTorch模型文件](model/pytorch/watermark-centernet)到本地/model/deps
1. 下载[待部署的PyTorch模型文件](model/pytorch/mnist)到本地/model/deps
下载[startPredict.sh](../../PyTorchPredictOnline/startPredict.sh), [gpu/predictor.py](../../PyTorchPredictOnline/gpu/predictor.py), [localtime](../../PyTorchPredictOnline/DockerImages/deps/localtime)到本地/model/deps
下载[Dockerfile_gpu](../../PyTorchPredictOnline/DockerImages/Dockerfile_gpu)到本地/model,并重命名为Dockerfile
2. 创建镜像,命令:
Expand Down
10 changes: 10 additions & 0 deletions DLPredictOnline/demo/model/pytorch/mnist/README.MD
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
# Mnist模型示例

client程序 [PyTorchClient](../../../src/main/java/com/bj58/ailab/demo/client/PyTorchClient.java)

## 数据
[test_data](./test_data) 样本数据

## 模型文件
以版本命名,所以模型下载需要解压成对应版本
[版本1](http://wos.58cdn.com.cn/nOlKjIhGntU/dlinference/demo_model_pytorch_mnist_2.zip)
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
18 changes: 0 additions & 18 deletions DLPredictOnline/demo/model/pytorch/watermark-centernet/README.MD

This file was deleted.

Binary file not shown.

This file was deleted.

Loading

0 comments on commit f4734be

Please sign in to comment.