Custom checkpoints step

[minhquoc0712]

Hi,

Thank you for releasing the code.

I'm not familiar with Ray. Do you know how to change the code to save the checkpoints over a certain number of training steps instead of epochs?