We test all of our tasks for compatibility with torch.jit. This table gives a breakdown of the supported features.
| Task | torch.jit.script |
torch.jit.trace |
torch.jit.save |
|---|---|---|---|
~flash.image.classification.model.ImageClassifier |
Yes | Yes | Yes |
~flash.image.detection.model.ObjectDetector |
Yes | No | Yes |
~flash.image.embedding.model.ImageEmbedder |
Yes | Yes | Yes |
~flash.image.segmentation.model.SemanticSegmentation |
No | Yes | Yes |
~flash.image.style_transfer.model.StyleTransfer |
No | Yes | Yes |
~flash.tabular.classification.model.TabularClassifier |
No | Yes | No |
~flash.text.classification.model.TextClassifier |
No | Yes * | Yes |
~flash.text.seq2seq.summarization.model.SummarizationTask |
No | Yes | Yes |
~flash.text.seq2seq.translation.model.TranslationTask |
No | Yes | Yes |
~flash.video.classification.model.VideoClassifier |
No | Yes | Yes |
* with strict=False