Commit Graph

2 Commits (5b2ee7267310c36a4c72e8e61e9b8322b7798623)

Author SHA1 Message Date
Ryan Nett 378669cc10
Fix MNIST Fetcher to not re-allocate each batch (#200)
* 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>
2019-08-29 21:28:34 -07:00
skymindops b5f0ec072f Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00