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.

Sign up for GitHub

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

Jump to bottom

Add auto batch size calculation in training scripts #17

Open
hvgazula opened this issue May 12, 2024 · 0 comments
Open

Add auto batch size calculation in training scripts #17

hvgazula opened this issue May 12, 2024 · 0 comments
Assignees
@hvgazula

Comments

@hvgazula
Copy link
Contributor

hvgazula commented May 12, 2024 •
edited
Loading

  • code snippet in test_train.py
  • implement it in brainy_train.py and kwyk_train.py as a decorator (maybe)
  • this is very useful for kwyk training which is trained on small blocks and can be inefficient with batch_size = 1
@hvgazula hvgazula self-assigned this May 12, 2024
@hvgazula hvgazula added this to Harsha May 12, 2024
@hvgazula hvgazula moved this to Todo in Harsha May 12, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@hvgazula hvgazula

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@hvgazula