When I update the DGL to latest version (0.3), I found the edge_softmax has been rewritten. However, this results the prediction of GAT become unstable with the same parameters (under version 0.2).
What is the difference between the Edge_softmax in 0.3 and 0.2?