port to new ray train api in estimators

oap-project / raydp

RayDP provides simple APIs for running Spark on Ray and integrating Spark with AI libraries.

Apache License 2.0

308 stars 68 forks source link

port to new ray train api in estimators #286

Closed kira-lin closed 1 year ago

kira-lin commented 1 year ago

Currently used ray.train.Trainer has been deprecated. In addition, the get_model method is not working now. This PR updates to use TorchTrainer and TensorflowTrainer and implements get_model based on checkpoint.

carsonwang commented 1 year ago

@KepingYan , can you please take a look as well?