diff --git a/pgl/layers/conv.py b/pgl/layers/conv.py index 96e1bb990a5b200efd6b360b7d4400cfa71db8dd..bbb364ab64a89f8ee3ee002cd8a9294a1710f804 100644 --- a/pgl/layers/conv.py +++ b/pgl/layers/conv.py @@ -230,7 +230,7 @@ def gin(gw, epsilon.stop_gradient = True msg = gw.send(send_src_copy, nfeat_list=[("h", feature)]) - output = gw.recv(msg, "sum") + (1.0 + epsilon) * feature + output = gw.recv(msg, "sum") + feature * (epsilon + 1.0) output = fluid.layers.fc(output, size=hidden_size,