-
Notifications
You must be signed in to change notification settings - Fork 1.5k
errer: tramsform Model to Torch Script #89
New issue
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
Comments
Hello! Yes, the latest update to the swish implementation saves memory but broke torchscript. An update to fix this is coming. |
Done -- check the latest update! |
thanks!! let me try. |
Do you get it done on this issue? I am precisely in the same trouble. My pytorch version is 1.3.1. |
Hi Luke! I have problems with scripting your implementation:
And with tracing: So, unfortunately, can't use the EN(((( |
I am having the same issue as @ne-bo, has there been any progress on this? Thanks so much! |
I am having the same issue too |
|
A workaround is to set |
Hi ! I have encountered the same problem stated at the beginning of this issue. When I try to get the torch script of my efficientnet model I get this error
This is my code
My pytorch version is 1.7.1 Thanks |
Than you @mzuzic for #89 (comment) |
when I want to save the trained Model as .pt. The error is (RuntimeError:
Could not export Python function call 'SwishImplementation'. Remove calls to Python functions before export. Did you forget add @script or @script_method annotation? If this is a nn.ModuleList, add it to constants:) . How could I solve it? thanks!!!
The text was updated successfully, but these errors were encountered: