You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@singa.apache.org by wa...@apache.org on 2016/08/14 09:11:44 UTC
[3/3] incubator-singa git commit: Merge PR #240 for training RBM
againt MNIST
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/dev
Commit: 1db278417bd5e569ef69a2aa83af26ee6f609701
Parents: dffae6b 5b332a4
Author: Wei Wang <wa...@gmail.com>
Authored: Sun Aug 14 17:10:37 2016 +0800
Committer: Wei Wang <wa...@gmail.com>
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(-)
----------------------------------------------------------------------