Hi,
You have a custom max pooling implementation in torch_geometric? The indexing error comes from the fact that during the backward of the pooling, the indices given to the ScatterMax Function are not correct.
The other one is some device assert that fails when calling the function SplineWeighting. I’m not sure what it does, it’s from your torch_spline_conv package, but something must be wrong with it.