[Dreambooth] Add a function to log interim generation #1711
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
This pull request allows to log interim generation during Dreambooth.
Usage
add an argument
--generating_progress_steps <int>. The process generates an interim image per steps you choose.To be able to generate deterministically, you can also set
--seedargument such as--seed 42.Generated images are stored at
f"{args.output_dir}/logs/.