textual_inversion icon indicating copy to clipboard operation
textual_inversion copied to clipboard

Config Parameter to Control the number of iterations?

Open 2454511550Lin opened this issue 1 year ago • 1 comments

Dear Author, I have a naive question, which parameter controls the number of iterations when training the embeddings? I learned from your paper that 5k is generally sufficient but I would like to do some experimenting myself.

I thought it is model.params.timesteps. I tried to change to 5, but the training did not stop after 5 epochs.

Any help would be greatly appreciated. Thank you!

2454511550Lin avatar Apr 03 '23 18:04 2454511550Lin