textual_inversion
textual_inversion copied to clipboard
Config Parameter to Control the number of iterations?
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!