File size: 3,199 Bytes
a01ef8c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
: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