search

Linaqruf / kohya-trainer

Adapted from https://note.com/kohya_ss/n/nbf7ce8d80f29 for easier cloning
Apache License 2.0
1.84k stars 302 forks source link

I don't know how to save checkpoint every n steps anymore in new colab #197

Open Michaelwhite34 opened 1 year ago

Linaqruf commented 1 year ago

which notebook? fyi we never have any parameter named save_every_n_steps image

MushroomFleet commented 1 year ago

Perhaps i misunderstood this value of 1 here? It only saves the CKPT after completing all steps, regardless of the total Epoch count. 7b7adc46d8ed75a08cb6712b776c661f

maybe its this? 6c7ae2044968aad91036ea4eea66a9e1

It would be useful to be able to save intermediate checkpoints on larger training sessions 👍 I usually Train LORA's with your notebooks, but I like using your Trainer for CKPT training lately.

Thanks for your hard work on it ! 💯

EDIT: I just set this value to 10 and it divided the Training Steps into 10 checkpoints for me, so easy, i just needed to do that! I solved my own problem :D