Merge PR #240 for training RBM againt MNIST
Project: http://git-wip-us.apache.org/repos/asf/incubator-singa/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-singa/commit/1db27841 Tree: http://git-wip-us.apache.org/repos/asf/incubator-singa/tree/1db27841 Diff: http://git-wip-us.apache.org/repos/asf/incubator-singa/diff/1db27841 Branch: refs/heads/master Commit: 1db278417bd5e569ef69a2aa83af26ee6f609701 Parents: dffae6b 5b332a4 Author: Wei Wang <[email protected]> Authored: Sun Aug 14 17:10:37 2016 +0800 Committer: Wei Wang <[email protected]> Committed: Sun Aug 14 17:10:37 2016 +0800 ---------------------------------------------------------------------- examples/mnist/README.md | 18 +++++ examples/mnist/train.py | 134 ++++++++++++++++++++++++++++++++ include/singa/core/tensor.h | 19 +++++ include/singa/model/loss.h | 1 - src/core/tensor/math_kernel.cu | 53 ++++++++++++- src/core/tensor/math_kernel.h | 14 +++- src/core/tensor/tensor.cc | 15 +++- src/core/tensor/tensor_math.h | 24 ++++++ src/core/tensor/tensor_math_cpp.h | 42 ++++++++++ src/core/tensor/tensor_math_cuda.h | 40 +++++++++- src/python/singa/optimizer.py | 7 +- src/python/singa/tensor.py | 20 ++++- src/python/swig/core_tensor.i | 10 +++ 13 files changed, 378 insertions(+), 19 deletions(-) ----------------------------------------------------------------------
