tch-rs icon indicating copy to clipboard operation
tch-rs copied to clipboard

Published 20 hours ago • verified publisher icon LaurentMazare

Using external modules like torch-scatter in JIT

Open KeKsBoTer opened this issue 3 years ago • 0 comments

Hi,

I want to load a JIT model with tch-rs but get the following error:

thread 'main' panicked at 'cannot load model: Torch("\nUnknown builtin op: torch_scatter::scatter_max.\nCould not find any similar ops to torch_scatter::scatter_max. This op may not exist or may not be currently supported in TorchScript

Is it possible to link the C++ version of pytorch-scatter to get this working?

KeKsBoTer avatar Jul 05 '22 08:07 KeKsBoTer