Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Support torch.jit.script for tasks #386

Closed
ethanwharris opened this issue Jun 9, 2021 · 0 comments · Fixed by #389
Closed

Support torch.jit.script for tasks #386

ethanwharris opened this issue Jun 9, 2021 · 0 comments · Fixed by #389
Assignees
@ethanwharris
Labels
enhancement New feature or request help wanted Extra attention is needed
Milestone
v0.3.x

Comments

@ethanwharris
Copy link
Collaborator

ethanwharris commented Jun 9, 2021 •
edited
Loading

Feature

Let's add support for torch.jit.script with tasks. Currently it gives an error. Using torch.jit.trace works fine.
@ethanwharris ethanwharris added bug / fix Something isn't working help wanted Extra attention is needed Priority labels Jun 9, 2021
@ethanwharris ethanwharris added this to the 0.3.x milestone Jun 9, 2021
@ethanwharris ethanwharris self-assigned this Jun 9, 2021
@ethanwharris ethanwharris changed the title Segmentation can't be jitted Tasks can't be jitted with torch.jit.script Jun 9, 2021
@ethanwharris ethanwharris added enhancement New feature or request and removed Priority bug / fix Something isn't working labels Jun 9, 2021
@ethanwharris ethanwharris changed the title Tasks can't be jitted with torch.jit.script Support torch.jit.script for tasks Jun 9, 2021
@ethanwharris ethanwharris mentioned this issue Jun 10, 2021
Merged
19 tasks
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees

@ethanwharris ethanwharris

Labels
enhancement New feature or request help wanted Extra attention is needed
Projects
None yet
Milestone
v0.3.x
Development

Successfully merging a pull request may close this issue.

Jit support Lightning-Universe/lightning-flash
1 participant
@ethanwharris