Flux LoRA resumes wrong epochs

[joneschunghk]

I am training a LoRA with 15 images. I saved a state on epoch 4. The train_state.json file inside x-0000004-state is {"current_epoch": 4, "current_step": 60}

And then I resume x-0000004-state on epoch 5, and saved a state on epoch 25. The train_state.json file inside x-0000025-state is {"current_epoch": 25, "current_step": 60}

And then I resume x-0000025-state on epoch 22, and saved a state on epoch 50. The train_state.json file inside x-0000050-state is {"current_epoch": 50, "current_step": 435}

And then I resume x-0000050-state on epoch 30. I will keep repeating the reverse training