Repository: incubator-singa Updated Branches: refs/heads/master 1cfdac6c3 -> a1553f845
SINGA-125 Improve Python Helper -Fix bug, add num_filter to field Project: http://git-wip-us.apache.org/repos/asf/incubator-singa/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-singa/commit/a1553f84 Tree: http://git-wip-us.apache.org/repos/asf/incubator-singa/tree/a1553f84 Diff: http://git-wip-us.apache.org/repos/asf/incubator-singa/diff/a1553f84 Branch: refs/heads/master Commit: a1553f845cdd0c5014caae1d315c0bcfcdf7aec5 Parents: 1cfdac6 Author: aaronwwf <[email protected]> Authored: Fri Jan 15 20:23:05 2016 +0800 Committer: aaronwwf <[email protected]> Committed: Fri Jan 15 20:23:05 2016 +0800 ---------------------------------------------------------------------- tool/python/singa/layer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-singa/blob/a1553f84/tool/python/singa/layer.py ---------------------------------------------------------------------- diff --git a/tool/python/singa/layer.py b/tool/python/singa/layer.py index 491e98b..079ce94 100644 --- a/tool/python/singa/layer.py +++ b/tool/python/singa/layer.py @@ -115,7 +115,7 @@ class Convolution2D(Layer): assert nb_filter > 0, 'nb_filter should be set as positive int' super(Convolution2D, self).__init__(name=generate_name('conv', 1), type=kCConvolution) - fields = {} + fields = {"num_filter":nb_filter} # for kernel if type(kernel) == int: fields['kernel'] = kernel
