-
Notifications
You must be signed in to change notification settings - Fork 8
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Exporting EgoNN model via TorchScript #4
Comments
Hi Milad,
I suspect that MinkowskiEngine library may not be compatible with
TorchScript. It contains low-level CUDA kernels - for efficient sparse
convolutions. So likely it cannot be converted to TorchScript. But I don't
know how to help.
regards
Jacek Komorowski
pon., 7 lis 2022 o 08:01 Milad Ramezani ***@***.***>
napisał(a):
… Dear Jacek,
I'd like to export the EgoNN model in TorchScript format.
At the moment, I get an error when loading the entire model. However, when
loading only the weights in torch.jit.script, there is no problem. My
impression is maybe TorchScript is not compatible with ME. If you have any
insights on this issue, I appreciate it if you share them with me.
Thanks!
—
Reply to this email directly, view it on GitHub
<#4>, or unsubscribe
<https://github.com/notifications/unsubscribe-auth/AASQZXX3EOARIRTRZ22KVQLWHCSNJANCNFSM6AAAAAARY3XCZM>
.
You are receiving this because you are subscribed to this thread.Message
ID: ***@***.***>
|
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Dear Jacek,
I'd like to export the EgoNN model in TorchScript format.
At the moment, I get an error when loading the entire model. However, when loading only the weights in
torch.jit.script
, there is no problem. My impression is maybe TorchScript is not compatible with ME. If you have any insights on this issue, I appreciate it if you share them with me.Thanks!
The text was updated successfully, but these errors were encountered: