eric-haibin-lin commented on issue #7034: How to switch a layer to cpu mode while the other layers are in gpu mode. URL: https://github.com/apache/incubator-mxnet/issues/7034#issuecomment-320524720 You can try the `ctx2group` option during simple_bind(). For example, if you want to specify a layer on cpu among the gpu layers, ``` a = mx.sym.var("a") b = mx.sym.var("b") with mx.AttrScope(ctx_group="cpu"): c = mx.sym.sample_negative_binomial(a,b) # runs on cpu d = mx.sym.dot(a, c) # runs on gpu executor = d.simple_bind(ctx=mx.gpu(), a=(10,), b=(10,), group2ctx={"cpu": mx.cpu()}) ``` allows `sample_negative_binomial` to be on cpu while the rest on gpu. Caveat: there's a known bug for elemwise_add with group2ctx https://github.com/apache/incubator-mxnet/issues/7080 ---------------------------------------------------------------- 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: us...@infra.apache.org
With regards, Apache Git Services