Closed Yard1 closed 2 years ago
Implements predict_proba, returning a ray.data.Dataset. Refactors generation of training functions. Adds ddp_kwargs argument. Improves checkpointing and adds an example.
predict_proba
ray.data.Dataset
ddp_kwargs
Implements
predict_proba, returning aray.data.Dataset. Refactors generation of training functions. Addsddp_kwargsargument. Improves checkpointing and adds an example.