singa-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From wang...@apache.org
Subject incubator-singa git commit: SINGA-125 Improve Python Helper
Date Sat, 16 Jan 2016 03:04:42 GMT
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 <dcswuw@gmail.com>
Authored: Fri Jan 15 20:23:05 2016 +0800
Committer: aaronwwf <dcswuw@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


Mime
View raw message