Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add tag dropout #20

Closed
wants to merge 1 commit into from
Closed

Add tag dropout #20

wants to merge 1 commit into from

Conversation

Omegastick
Copy link

@Omegastick Omegastick commented Dec 28, 2022

A small amount (0.01, for example) of tag dropout can be useful for very thoroughly-tagged datasets to stop the model overfitting to the the tagger. It's particularly useful with the WD 1.4 tagger.

@bmaltais
Copy link
Owner

This look really cool... but I don't maintain the source code for the tool itself, just for the GUI. I suggest you propose this pull request directly on kohya_ss repo where the actual code is maintained... yeah... confusing... sorry.

Here is where you should submit the PR and once kohya merge it in his repo I will update mine with the feature: https://github.com/kohya-ss/sd-scripts

@Omegastick
Copy link
Author

Sure, I'll close this PR.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants