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.
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 prepareisn'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.