High: It blocks me to complete my task. I want to achieve following two steps while training a Tensorflow model using Ray train: Save the checkpoint with model weights in a pickle file Use this saved checkpoint pickle file later on to load the model with model weights. (For prediction or retrain…

Save and reuse Checkpoints in Ray 2.0 version

xwjiang2010 November 7, 2022, 10:01pm 2

Hi,
This is provided now in TensorflowCheckpoint class now. Ray AIR API — Ray 2.0.1

Not only can you save weights, you can also save and load saved_model and h5 formats as well.

[Train] How to use TensorflowCheckpoint to save model/h5

Topic		Replies	Views
Tuning a Keras model - no checkpoints saved Ray Tune	7	1512	March 1, 2023
How to save models after Trainer.fit()?	1	665	August 19, 2023
Save model parameters on each checkpoint Ray Tune	21	3380	March 29, 2023
Saving ray model to tf/pytorch Checkpointing, Restoring	0	301	August 11, 2023
How to make checkpoint by ray.tune.run and load it? RLlib	3	2789	July 7, 2022