Filed while practicing SSE

I have problems while running the code for Stochastic steady-state embedding (SSE).

the error is that-AttributeError: ‘NDArray’ object has no attribute ‘device’

I know that this is because I am using mxnet rather than pytorch. But how can I make it suitable for mxnet tensor?

Please set environment variable DGLBACKEND to mxnet ( and see if that helps,

Thanks for your help ! It’s workable.