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

[Feature Request]: Remove the hard coded batch size & add it to the advanced options #181

Open
Vigilence opened this issue Oct 7, 2024 · 0 comments

Comments

@Vigilence
Copy link

If the hard coded batch size can instead be edited in the advanced config that would be great. I figured out, after some testing that the batch size is hard coded in app.py file. I had initial tried changing the batch size via the advanced commands, but noticed it didn't increase vram usage or change the speed of the training.

So to actually change the batch size I:

  • Tried the advanced command "--train_batch_size", which doesn't work since its overridden by the hard coded batch size
  • Edited the hard coded batch size to what I needed, and noticed it changed the training parameters and actually worked (used more vram & visible difference in training speed).

Screenshot 2024-10-07 002721
Screenshot 2024-10-07 121154
Screenshot 2024-10-07 121238
Screenshot 2024-10-07 121700
Screenshot 2024-10-07 121713

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

No branches or pull requests

1 participant