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: Fix a typo for cudnn_prefer and change the default workspace limit size to 1024MB
Date Tue, 02 May 2017 09:07:55 GMT
Repository: incubator-singa
Updated Branches:
  refs/heads/master e527db986 -> 2d1dd4290


Fix a typo for cudnn_prefer and change the default workspace limit size to 1024MB


Project: http://git-wip-us.apache.org/repos/asf/incubator-singa/repo
Commit: http://git-wip-us.apache.org/repos/asf/incubator-singa/commit/2d1dd429
Tree: http://git-wip-us.apache.org/repos/asf/incubator-singa/tree/2d1dd429
Diff: http://git-wip-us.apache.org/repos/asf/incubator-singa/diff/2d1dd429

Branch: refs/heads/master
Commit: 2d1dd42901e25f42b95f80a3e0aea13d72e449e0
Parents: e527db9
Author: wangwei <wangwei@comp.nus.edu.sg>
Authored: Tue Apr 18 11:39:38 2017 +0800
Committer: wangwei <wangwei@comp.nus.edu.sg>
Committed: Tue May 2 17:07:21 2017 +0800

----------------------------------------------------------------------
 python/singa/layer.py | 8 ++++----
 src/proto/model.proto | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/incubator-singa/blob/2d1dd429/python/singa/layer.py
----------------------------------------------------------------------
diff --git a/python/singa/layer.py b/python/singa/layer.py
index 2ce204b..caafef0 100644
--- a/python/singa/layer.py
+++ b/python/singa/layer.py
@@ -306,7 +306,7 @@ class Conv2D(Layer):
             'same' -> padding is half of the kernel (floor), the kernel must be
             odd number.
         cudnn_prefer (string): the preferred algorithm for cudnn convolution
-            which could be 'fatest', 'autotune', 'limited_workspace' and
+            which could be 'fastest', 'autotune', 'limited_workspace' and
             'no_workspace'
         workspace_byte_limit(int): max workspace size in MB (default is 512MB)
         data_format (string): either 'NCHW' or 'NHWC'
@@ -330,7 +330,7 @@ class Conv2D(Layer):
     """
 
     def __init__(self, name, nb_kernels, kernel=3, stride=1, border_mode='same',
-                 cudnn_prefer='fatest', workspace_byte_limit=512,
+                 cudnn_prefer='fastest', workspace_byte_limit=1024,
                  data_format='NCHW', use_bias=True, W_specs=None, b_specs=None,
                  pad=None, input_sample_shape=None):
         super(Conv2D, self).__init__(name)
@@ -376,8 +376,8 @@ class Conv1D(Conv2D):
     """
 
     def __init__(self, name, nb_kernels, kernel=3, stride=1,
-                 border_mode='same', cudnn_prefer='fatest',
-                 workspace_byte_limit=512,
+                 border_mode='same', cudnn_prefer='fastest',
+                 workspace_byte_limit=1024,
                  use_bias=True, W_specs={'init': 'Xavier'},
                  b_specs={'init': 'Constant', 'value': 0}, pad=None,
                  input_sample_shape=None):

http://git-wip-us.apache.org/repos/asf/incubator-singa/blob/2d1dd429/src/proto/model.proto
----------------------------------------------------------------------
diff --git a/src/proto/model.proto b/src/proto/model.proto
index d4b85b6..822fb5c 100644
--- a/src/proto/model.proto
+++ b/src/proto/model.proto
@@ -389,7 +389,7 @@ message ConvolutionConf {
 
   // SINGA: add by xiangrui
   // cudnn workspace size in MB
-  optional int32 workspace_byte_limit = 50 [default = 512];
+  optional int32 workspace_byte_limit = 50 [default = 1024];
   // cudnn algorithm preference
   // options: "fastest", "limited_workspace", "no_workspace"
   optional string prefer = 51 [default = "fastest"];


Mime
View raw message