-
Notifications
You must be signed in to change notification settings - Fork 6.8k
Conversation
devtype2str = {1: 'cpu', 2: 'gpu'} | ||
devstr2type = {'cpu': 1, 'gpu': 2} | ||
devtype2str = {1: 'cpu', 2: 'gpu', 3: 'cpu_pinned'} | ||
devstr2type = {'cpu': 1, 'gpu': 2, 'cpu_pinned': 3} |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@tqchen please double check if it is ok to have cpu_pinned
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Let us not expose this for now
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
must be here, create state may create pinned mem array. That bug costs me a
hour to find
On Sun, Oct 11, 2015 at 1:45 AM Tianqi Chen [email protected]
wrote:
In python/mxnet/context.py
#263 (comment):@@ -29,8 +29,8 @@ class Context(object):
"""
# static class variable
default_ctx = None
- devtype2str = {1: 'cpu', 2: 'gpu'}
- devstr2type = {'cpu': 1, 'gpu': 2}
- devtype2str = {1: 'cpu', 2: 'gpu', 3: 'cpu_pinned'}
- devstr2type = {'cpu': 1, 'gpu': 2, 'cpu_pinned': 3}
Let us not expose this for now
—
Reply to this email directly or view it on GitHub
https://github.com/dmlc/mxnet/pull/263/files#r41705789.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I see what you mean. Keep it here then
@mli Please test all the modes in kvstore |
/** | ||
* \brief return the type | ||
*/ | ||
inline std::string type() { return type_; } |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This is very bad, as you return the space get destructed, and python side get a invalid space pointer. Return const std::string & instead
Three comments in general
|
On updater index hack for So in future we can use index to tell which weight is which(one reason we created it in this way) |
@@ -148,8 +148,7 @@ def create(name, rescale_grad=1, **kwargs): | |||
else: | |||
raise ValueError('Cannot find optimizer %s' % name) | |||
|
|||
|
|||
def optimizer_clossure(optimizer): | |||
def get_updater(optimizer): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
create is a better name, as it creates a updater closure each time
std::min requires algorithm header as per the standard. It seems to work without it on Linux/OS X but breaks on Windows.
* cub 89de7ab2(89de7ab2)...05eb57fa(05eb57fa) (798 commits) > Merge pull request sociomantic-tsunami#1 from ptrendx/update > Create README.md > update readme.md > 1.7.0 < 1.6.4 doc update (part 2) (...) * dlpack ()...a6e09b5(a6e09b5) (1 commits) > Change order of device_type/id in Context (sociomantic-tsunami#11) * dmlc-core a6c5701(a6c5701)...87b7ffa(87b7ffa) (54 commits) > add SetEnv (apache#322) > Fix a bug in seek/tell on Windows (apache#318) > Fixes apache#303: added recurse_directories to InputSplit::Create (apache#310) > Type name error (apache#316) > Small param bug (apache#315) (...) * mshadow c037b06(c037b06)...2d7780c(2d7780c) (42 commits) > [CMAKE][ARM] Change USE_SSE to SUPPORT_MSSE2 to it uses the autodetected presence of sse compiler flag from the parent project (see PR apache#8395) (apache#303) > Makes repeated setting of gpu rng seed produce repeatable sequences. (apache#304) > Add USE_SSE which propagates into MSHADOW_USE_SSE in cmake (apache#302) > fix range (apache#301) > fix for random seed generation (apache#300) (...) * nnvm b279286(b279286)...e4a138a(e4a138a) (139 commits) > [TVM] upgrade to latest version (apache#263) > Added support for CoreML Permute layers (apache#262) > [CMPL] Add Support for Other Data Types (apache#252) > fix onnx conv2d_transpose loading (apache#245) > [FIX] Fix from_mxnet for multiple outputs symbol (apache#247) (...) * ps-lite v1+118(acdb698)...v1+123(2ce8b9a) (2 commits) > Merge pull request apache#117 from madjam/listen-interface > Merge pull request apache#109 from b0noI/master
Fixes sociomantic-tsunami#11 * cub 89de7ab2(89de7ab2)...05eb57fa(05eb57fa) (798 commits) > Merge pull request sociomantic-tsunami#1 from ptrendx/update > Create README.md > update readme.md > 1.7.0 < 1.6.4 doc update (part 2) (...) * dlpack ()...a6e09b5(a6e09b5) (1 commits) > Change order of device_type/id in Context (sociomantic-tsunami#11) * cub 89de7ab2(89de7ab2)...05eb57fa(05eb57fa) (798 commits) > Merge pull request sociomantic-tsunami#1 from ptrendx/update > Create README.md > update readme.md > 1.7.0 < 1.6.4 doc update (part 2) (...) * dlpack ()...a6e09b5(a6e09b5) (1 commits) > Change order of device_type/id in Context (sociomantic-tsunami#11) * dmlc-core a6c5701(a6c5701)...87b7ffa(87b7ffa) (54 commits) > add SetEnv (apache#322) > Fix a bug in seek/tell on Windows (apache#318) > Fixes apache#303: added recurse_directories to InputSplit::Create (apache#310) > Type name error (apache#316) > Small param bug (apache#315) (...) * mshadow c037b06(c037b06)...2d7780c(2d7780c) (42 commits) > [CMAKE][ARM] Change USE_SSE to SUPPORT_MSSE2 to it uses the autodetected presence of sse compiler flag from the parent project (see PR apache#8395) (apache#303) > Makes repeated setting of gpu rng seed produce repeatable sequences. (apache#304) > Add USE_SSE which propagates into MSHADOW_USE_SSE in cmake (apache#302) > fix range (apache#301) > fix for random seed generation (apache#300) (...) * nnvm b279286(b279286)...e4a138a(e4a138a) (139 commits) > [TVM] upgrade to latest version (apache#263) > Added support for CoreML Permute layers (apache#262) > [CMPL] Add Support for Other Data Types (apache#252) > fix onnx conv2d_transpose loading (apache#245) > [FIX] Fix from_mxnet for multiple outputs symbol (apache#247) (...) * ps-lite v1+118(acdb698)...v1+123(2ce8b9a) (2 commits) > Merge pull request apache#117 from madjam/listen-interface > Merge pull request apache#109 from b0noI/master
replaced
kvstore_type
andudpate_on_server
bykvstore
(either a type or a KVStore) on model.py, more dochttps://github.com/mli/mxnet/blob/master/doc/developer-guide/multi_node.md
passed tests on mnist and cifar with multi-gpus