:orphan:

Image Classification Methods
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. currentmodule:: tlt.models.image_classification

.. autosummary::
  :toctree: _autosummary
  :nosignatures:
   
   tfhub_image_classification_model.TFHubImageClassificationModel.train
   tfhub_image_classification_model.TFHubImageClassificationModel.quantize
   tfhub_image_classification_model.TFHubImageClassificationModel.optimize_graph
   tfhub_image_classification_model.TFHubImageClassificationModel.benchmark

   tf_image_classification_model.TFImageClassificationModel.train
   tf_image_classification_model.TFImageClassificationModel.quantize
   tf_image_classification_model.TFImageClassificationModel.optimize_graph
   tf_image_classification_model.TFImageClassificationModel.benchmark

   keras_image_classification_model.KerasImageClassificationModel.train
   keras_image_classification_model.KerasImageClassificationModel.quantize
   keras_image_classification_model.KerasImageClassificationModel.optimize_graph
   keras_image_classification_model.KerasImageClassificationModel.benchmark

   torchvision_image_classification_model.TorchvisionImageClassificationModel.train
   torchvision_image_classification_model.TorchvisionImageClassificationModel.quantize
   torchvision_image_classification_model.TorchvisionImageClassificationModel.benchmark

   pytorch_image_classification_model.PyTorchImageClassificationModel.train
   pytorch_image_classification_model.PyTorchImageClassificationModel.quantize
   pytorch_image_classification_model.PyTorchImageClassificationModel.benchmark

   pytorch_hub_image_classification_model.PyTorchHubImageClassificationModel.train
   pytorch_hub_image_classification_model.PyTorchHubImageClassificationModel.quantize
   pytorch_hub_image_classification_model.PyTorchHubImageClassificationModel.benchmark

   image_classification_model.ImageClassificationModel.train
   image_classification_model.ImageClassificationModel.quantize
   image_classification_model.ImageClassificationModel.benchmark

Text Classification Methods
^^^^^^^^^^^^^^^^^^^^^^^^^^^^

.. currentmodule:: tlt.models.text_classification

.. autosummary::
  :toctree: _autosummary
  :nosignatures:
  :recursive:

   tf_text_classification_model.TFTextClassificationModel.train
   tf_text_classification_model.TFTextClassificationModel.quantize
   tf_text_classification_model.TFTextClassificationModel.optimize_graph
   tf_text_classification_model.TFTextClassificationModel.benchmark

   pytorch_hf_text_classification_model.PyTorchHFTextClassificationModel.train
   pytorch_hf_text_classification_model.PyTorchHFTextClassificationModel.quantize
   pytorch_hf_text_classification_model.PyTorchHFTextClassificationModel.benchmark

   tf_hf_text_classification_model.TFHFTextClassificationModel.train
   tf_hf_text_classification_model.TFHFTextClassificationModel.quantize
   tf_hf_text_classification_model.TFHFTextClassificationModel.optimize_graph
   tf_hf_text_classification_model.TFHFTextClassificationModel.benchmark
   
   text_classification_model.TextClassificationModel.train
   text_classification_model.TextClassificationModel.quantize
   text_classification_model.TextClassificationModel.benchmark