Scaling of edge network with 0.1

[tsjain]

I did not see where the edge network weights are scaled by 0.1 in the code.

[Lin-Yijie]

Well, we do not implement the Glorot initialization with a scaling factor of 0.1. Instead, we directly use the default initialization of torch. If you figure out how to change this, you could provide a pull request.