Support for PyTorch model (#346)

* Add files via upload * Add files via upload * Add files via upload * Update build_docker_images.sh * Update pytorch.py * Update pytorch_container.py * Add files via upload * Add files via upload * Create deploy_pytorch_models.py * Update deploy_pytorch_models.py * Update deploy_pytorch_models.py * Update deploy_pytorch_models.py * Update deploy_pytorch_models.py * Update deploy_pytorch_models.py * Update pytorch.py * Update deploy_pytorch_models.py * Update deploy_pytorch_models.py * Update deploy_pytorch_models.py * Update pytorch_container.py * Update deploy_pytorch_models.py * Update deploy_pytorch_models.py * Capture class dependencies * remove prints * Format code * Address review comments * Format code * Fix container imports * Correct print statement typo in closure container * Format code * Don't install clipper_admin where only cloudpickle is needed * Add missing cloudpickle dependency in tests file
ucbrise · Jan 10, 2018 · 3b44a91 · 3b44a91
1 parent 193cf8b
commit 3b44a91
Show file tree

Hide file tree

Showing 16 changed files with 581 additions and 901 deletions.
diff --git a/bin/build_docker_images.sh b/bin/build_docker_images.sh
@@ -258,6 +258,7 @@ build_images () {
     create_image pyspark-container PySparkContainerDockerfile $public
     create_image tf_cifar_container TensorFlowCifarDockerfile $public
     create_image tf-container TensorFlowDockerfile $public
+    create_image pytorch-container PyTorchContainerDockerfile $public
 }