zilunpeng/use_ray_to_get_pred.py

Created March 23, 2021 22:22

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/zilunpeng/e629c0270d7c5fef8d579d72b48f08b1.js"></script>
Save zilunpeng/e629c0270d7c5fef8d579d72b48f08b1 to your computer and use it in GitHub Desktop.

Download ZIP

Get all predictions using Ray. Code below is part of the distributed inference notebook (https://git.io/JYeQQ).

Raw

use_ray_to_get_pred.py

predictions = ray.get(prediction_futures)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment