huyangc opened a new issue #14761: mutable state cannot go across device 
op=BatchNorm input_state_index=3
URL: https://github.com/apache/incubator-mxnet/issues/14761
 
 
   Using the same code as #7034
   ```
   data = mx.sym.var('data')
   with mx.AttrScope(ctx_group="cpu"):
       c = mx.sym.Convolution(data) # runs on cpu
   d = mx.sym.BatchNorm(c) # runs on gpu
   
   executor = d.simple_bind(ctx=mx.gpu(), group2ctx={"cpu": mx.cpu()})
   ```
   got this error mutable state cannot go across device op=BatchNorm 
input_state_index=3

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to 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

Reply via email to