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 <[email protected]> Authored: Tue Apr 18 11:39:38 2017 +0800 Committer: wangwei <[email protected]> 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"];
