Scalability of Pytorch version

#1

Hello! Congratulations on the impressive library. I notice that the main example you give of training on very large graph is an mxnet implementation of stochastic steady state embedding, but that this example is not implemented in Pytorch. Is this because the pytorch version is not as scalable as the mxnet version? Or would be possible to run a pytorch version of stochastic steady state embedding on a similarly large graph and get performance of the same order of magnitude?

0 Likes

#2

That’s a good question. We are working on an example for large graph training using graph sampling method, which we think is more useful for real scenarios. For the up-coming example, we will release both Pytorch and MXNet versions.

0 Likes

#3

Thanks. Look forward to seeing the new example.

0 Likes