torchexpo.modules¶
-
class
torchexpo.modules.
ImageClassificationModule
(model, model_name, model_example)[source]¶ Image Classification module for all vision models under vision/image_classification
- Parameters
model – Pretrained model which is used for conversion
model_name – Name of the model
model_example – Example tensor used as example input while extraction
-
class
torchexpo.modules.
ImageSegmentationModule
(model, model_name, model_example)[source]¶ Image Segmentation module for all vision models under vision/image_segmentation
- Parameters
model – Pretrained model which is used for conversion
model_name – Name of the model
model_example – Example tensor used as example input while extraction
-
class
torchexpo.modules.
SentimentAnalysisModule
(model, model_name, model_example)[source]¶ Sentiment Analysis module for all nlp models under nlp/sentiment_analysis
- Parameters
model – Pretrained model which is used for conversion
model_name – Name of the model
model_example – Example tensor used as example input while extraction
-
class
torchexpo.modules.
TorchExpoModule
(model, model_name, model_example)[source]¶ Base class for all PyTorch models
- Parameters
model – Pretrained model which is used for conversion
model_name – Name of the model
model_example – Example tensor used as example input while extraction