Commit Graph

2 Commits (a6de3b5d6fd92a5de7c72ea522d36fa9544957a3)

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