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 2017/05/02 09:07:55 UTC
incubator-singa git commit: Fix a typo for cudnn_prefer and change
the default workspace limit size to 1024MB
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 <wa...@comp.nus.edu.sg>
Authored: Tue Apr 18 11:39:38 2017 +0800
Committer: wangwei <wa...@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"];