How can I convert a model to a pre-trained model to have 0 step?

[hyunR]

Hi all!

I have trained a model and I'd like to use it as a base model with different data set.

So I thought I can only change the file name from D_xxx -> D_0 and use it as a pre-trained model.

I found no issue with the training, but found that the "steps" information from the pre-trained model got inherited to the new training. For example, if I have 50k steps from pre-trained then the new trained one(with the renamed D_0 as a pre-trained model) starts at 50k steps.

Does anyone know how to properly create a pre-trained model? I have searched other Github repos and Huggingface spaces but could not find a way...