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

Fix MNIST Fetcher to not re-allocate each batch #200

Merged
merged 3 commits into from Aug 30, 2019

Conversation

@rnett
Copy link
Collaborator

commented Aug 30, 2019

Fixes eclipse#7944.

rnett added 3 commits Aug 30, 2019
don't allocate so many float arrays, use INDArrays instead
Signed-off-by: Ryan Nett <rnett@skymind.io>
re-add pre-processing, better names
Signed-off-by: Ryan Nett <rnett@skymind.io>
use float[][] pool to avoid extra ndarray creation
Signed-off-by: Ryan Nett <rnett@skymind.io>

@rnett rnett requested a review from AlexDBlack Aug 30, 2019

@rnett rnett self-assigned this Aug 30, 2019

@AlexDBlack
Copy link

left a comment

LGTM 👍

@rnett rnett merged commit 378669c into master Aug 30, 2019

@rnett rnett deleted the rn_mnist_fetcher_fix branch Aug 30, 2019

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
2 participants
You can’t perform that action at this time.