I have an issue with not being able to reproduce the same output with the same input using the propagate method from the MessagePassing class from torch_geometric.nn (if it matters, aggr=‘mean’). I’ve narrowed it down to being that method specifically, by checking that everything in my own message method is deterministic. I’m otherwise using torch.manual_seed, but there seems to be some randomness induced by that specific method.
My question then follows:
Why is there even randomness in that method to begin with? Am I missing something, or isn’t it desired to have the same aggregated message for the same input?
How can I overcome this, as it is inducing problems for my downstream task?