about onnx #259
Description
I use fps from torch_cluster in my model. When I convert the model to ONNX, I get:
torch.onnx.errors.UnsupportedOperatorError: ONNX export failed on an operator with unrecognized namespace torch_cluster::fps.
If you are trying to export a custom operator, make sure you registered it with the right domain and version.
What can I do? My model sometimes needs to run on CPU, and I’d like to keep using torch_cluster’s fps instead of my own handwritten version, which is much slower.