You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by GitBox <gi...@apache.org> on 2017/12/06 07:08:45 UTC
[GitHub] NonvolatileMemory opened a new issue #8963: set_data() get wrong with gpu
NonvolatileMemory opened a new issue #8963: set_data() get wrong with gpu
URL: https://github.com/apache/incubator-mxnet/issues/8963
Env:mxnet-cu80 1.0
this code is ok on cpu,but with gpu it will crash.
SMN = SMN_Last()
SMN.initialize(ctx=ctx)
word2vec = (nd.array(wordvecs.W)).as_in_context(mx.gpu())
print(word2vec)
SMN.Embed.weight.set_data(word2vec)
I got this error report:
RuntimeError: Parameter smn_last0_embedding0_weight was not initialized on context cpu(0). It was only initialized on [gpu(0)]
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
With regards,
Apache Git Services