wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-09-12 12:51:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Tune] Fix tune serve integration example (#13233)

Browse Source
This commit is contained in:
Kai Fricke
2021-01-06 17:02:04 +01:00
committed by GitHub
parent 32dc5676b4
commit 97211a6170
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
doc/source/tune/_tutorials/tune-serve-integration-mnist.py
+2 -2
View File
@@ -440,7 +440,7 @@ class MNISTBackend:
#######################################################################
# We would like to have a fixed location where we store the currently
# active model. We call this directory ``model_dir``. Everytime we
# active model. We call this directory ``model_dir``. Every time we
# would like to update our model, we copy the checkpoint of the new
# model to this directory. We then create a new backend pointing to
# that directory, route all the traffic on our model endpoint to this
@@ -615,7 +615,7 @@ if __name__ == "__main__":
args = parser.parse_args()
if args.smoke_test:
ray.init(num_cpus=2)
ray.init(num_cpus=3)
model_dir = os.path.expanduser(args.model_dir)