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

Support dynamic dimensions #209

Open
BinaryWarlock opened this issue Apr 9, 2024 · 0 comments
Open

Support dynamic dimensions #209

BinaryWarlock opened this issue Apr 9, 2024 · 0 comments

Comments

@BinaryWarlock
Copy link

Is your feature request related to a problem? Please describe.
I'd like to use dynamic batch sizes to actually batch properly.

Describe the solution you'd like
Dynamic dimension support

Describe alternatives you've considered
Baking in the batch size with nnx prepare isn't remotely ideal, since if my dataset is not a multiple of the batch size I'd have to pad it somehow and that's pretty awful.

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