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: [email protected] With regards, Apache Git Services
