Matthias Fey comments

Results 1128 comments of


                                            Matthias Fey

Request to add two graph self-supervised methods

That sounds cool :) We are very happy to take a pull request in on that one, including both the model and an example. Let me know if you have...

Add `nn.models.GRetriever`

Very clean now, thanks for the updates :)

Err: source code compile

Can you share a bit more information about your system (OS, Python PyTorch version)?

pip install not working with torch==2.3.0

If you are switching between PyTorch versions, did you make sure to correctly uninstall all existing installations of `torch-cluster` and re-install them? Usually, an error like `Symbol not found: __ZN5torch3jit17parseSchemaOrNameERKNSt3__112basic_stringIcNS1_11char_traitsIcEENS1_9allocatorIcEEEEb`...

how to implement cdist

Interesting problem. The best way to implement this is indeed with a custom CUDA kernel, in which way you can drop the for-loop in option2/3 and avoid the memory blowup...

Wheels for pytorch 2.5.1

You can use the wheels from 2.5.0

uv add torch_cluster fails due to undeclared dependency on torch

you can install with `--no-isolation`

Integration with DeepSpeed: expected scalar type Float but found Half

Thanks for reporting, will look into it.

segment_csr triggers shape error in backward pass when indptr does on cover src and reduction is mean

Do you mean `indptr = torch.tensor([0, 3, 8])`?