databricks / spark-deep-learning

Deep Learning Pipelines for Apache Spark
https://databricks.github.io/spark-deep-learning
Apache License 2.0
2k stars 494 forks source link

[ML-6097] Update Horovod Runner to include return value and multi-gpu info. #183

Closed MrBago closed 5 years ago

MrBago commented 5 years ago

Cherry-picked the commit that adds return value implemenation, docs & tests. Added better description of np to include info about multi-gpu support.

mengxr commented 5 years ago

LGTM