Repository: incubator-singa Updated Branches: refs/heads/master 1e30a8b73 -> 2aa1d2241
fix bugs in layer::param_names() and tensor::sum_columns() (device was not set) Project: http://git-wip-us.apache.org/repos/asf/incubator-singa/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-singa/commit/2aa1d224 Tree: http://git-wip-us.apache.org/repos/asf/incubator-singa/tree/2aa1d224 Diff: http://git-wip-us.apache.org/repos/asf/incubator-singa/diff/2aa1d224 Branch: refs/heads/master Commit: 2aa1d22416792507d7ddf7c53390bd86535dafa8 Parents: 1e30a8b Author: Wei Wang <[email protected]> Authored: Mon Apr 3 19:50:18 2017 +0800 Committer: Wei Wang <[email protected]> Committed: Mon Apr 3 19:50:18 2017 +0800 ---------------------------------------------------------------------- python/singa/layer.py | 2 +- python/singa/tensor.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-singa/blob/2aa1d224/python/singa/layer.py ---------------------------------------------------------------------- diff --git a/python/singa/layer.py b/python/singa/layer.py index 6302b45..b985783 100644 --- a/python/singa/layer.py +++ b/python/singa/layer.py @@ -168,7 +168,7 @@ class Layer(object): ''' names = [] for x in self.param_specs: - names.append(x['name']) + names.append(x.name) return names def param_values(self): http://git-wip-us.apache.org/repos/asf/incubator-singa/blob/2aa1d224/python/singa/tensor.py ---------------------------------------------------------------------- diff --git a/python/singa/tensor.py b/python/singa/tensor.py index 82361c5..6d8abf3 100644 --- a/python/singa/tensor.py +++ b/python/singa/tensor.py @@ -1046,7 +1046,7 @@ def sum_columns(M): a new Tensor as the resulted column. ''' assert M.ndim() == 2, 'M.nDim() is supposed to be 2' - ret = Tensor((M.shape[0], 1)) + ret = Tensor((M.shape[0], 1), M.signa_tensor.device()) singa.SumColumns(M.singa_tensor, ret.singa_tensor) return ret @@ -1061,7 +1061,7 @@ def sum_rows(M): a new Tensor as the resulted row. ''' assert M.ndim() == 2, 'M.nDim() is supposed to be 2' - ret = Tensor((1, M.shape[1])) + ret = Tensor((1, M.shape[1]), M.singa_tensor.device()) singa.SumRows(M.singa_tensor, ret.singa_tensor) return ret
