Skip to content
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

Make all Flambe base-models torchscript compatible #221

Open
cle-ros opened this issue Jan 29, 2020 · 1 comment
Open

Make all Flambe base-models torchscript compatible #221

cle-ros opened this issue Jan 29, 2020 · 1 comment
Labels
enhancement New feature or request

Comments

@cle-ros
Copy link
Collaborator

cle-ros commented Jan 29, 2020

It would be great if we could extract torchscript models from flambe models, after training. That requires all flambe base models to be compatible with torchscript. After a very superficial attempt to make this work, I found that (at least) some typing instructions in flambe.component are torchscript incompatible.

@cle-ros cle-ros added the enhancement New feature or request label Jan 29, 2020
@jeremyasapp
Copy link
Contributor

100%. We should add a test suite to make sure that is the case, always

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants