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/01/16 04:04:42 UTC
incubator-singa git commit: SINGA-125 Improve Python Helper
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 <dc...@gmail.com>
Authored: Fri Jan 15 20:23:05 2016 +0800
Committer: aaronwwf <dc...@gmail.com>
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