search

oap-project / raydp

RayDP provides simple APIs for running Spark on Ray and integrating Spark with AI libraries.
Apache License 2.0
293 stars 66 forks source link

upgrade Checkpoint usage to fit ray 3.0 #388

Closed harborn closed 2 months ago

harborn commented 7 months ago

replace Checkpoint usage by using TorchCheckpoint/TensorflowCheckpoint

pang-wu commented 6 months ago

Test failed: https://github.com/oap-project/raydp/actions/runs/6779351884/job/18426275758?pr=388