lichen11 commented on issue #5296: KVStore error with certain optimizer settings - R-package URL: https://github.com/apache/incubator-mxnet/issues/5296#issuecomment-461562885 Hi, I am still encountering the problem of training mnist with multiple GPU. model <- mx.model.FeedForward.create(lenet, X=train.array, y=train.y, ctx=list(mx.gpu(0),mx.gpu(1)), num.round=100, array.batch.size=100, learning.rate=0.05, momentum=0.9, wd=0.00001, eval.metric=mx.metric.accuracy, epoch.end.callback=mx.callback.log.train.metric(100), kvstore = "device") The error I am getting is Start training with 2 devices Error in kvstore$set.optimizer(optimizer) : kvstore.cc:124: RCheck failed: names.size() == 2 && names[0] == "create.state" && names[1] == "update" Invalid optimizer I am using mxnet 1.5.0 obtained from github incubator-mxnet today.
---------------------------------------------------------------- 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
