Index

Symbols

--accuracy-criterion tlt-quantize command line option --accuracy_criterion tlt-quantize command line option --add-aug tlt-train command line option --add_aug tlt-train command line option --approach tlt-quantize command line option --batch-size tlt-benchmark command line option tlt-quantize command line option --batch_size tlt-benchmark command line option tlt-quantize command line option --class-names tlt-eval command line option tlt-train command line option --class_names tlt-eval command line option tlt-train command line option --context-key tlt-train command line option --context_key tlt-train command line option --cutpaste tlt-train command line option --dataset-catalog tlt-eval command line option tlt-train command line option --dataset-dir tlt-benchmark command line option tlt-eval command line option tlt-quantize command line option tlt-train command line option --dataset-file tlt-benchmark command line option tlt-eval command line option, [1] tlt-quantize command line option tlt-train command line option --dataset-name tlt-eval command line option tlt-train command line option --dataset_catalog tlt-eval command line option tlt-train command line option --dataset_dir tlt-benchmark command line option tlt-eval command line option tlt-quantize command line option tlt-train command line option --dataset_file tlt-benchmark command line option tlt-eval command line option, [1] tlt-quantize command line option tlt-train command line option --dataset_name tlt-eval command line option tlt-train command line option --delimiter tlt-benchmark command line option tlt-eval command line option tlt-quantize command line option tlt-train command line option --distributed tlt-train command line option --early-stopping tlt-train command line option --early_stopping tlt-train command line option --epochs tlt-train command line option --framework tlt-list-models command line option tlt-train command line option --hostfile tlt-train command line option --hvd-start-timeout tlt-train command line option --hvd_start_timeout tlt-train command line option --init-checkpoints tlt-train command line option --init_checkpoints tlt-train command line option --instruction-key tlt-train command line option --instruction_key tlt-train command line option --ipex-optimize tlt-train command line option --ipex_optimize tlt-train command line option --lora-alpha tlt-train command line option --lora-dropout tlt-train command line option --lora-rank tlt-train command line option --lora_alpha tlt-train command line option --lora_dropout tlt-train command line option --lora_rank tlt-train command line option

--lr-decay tlt-train command line option --lr_decay tlt-train command line option --max-new-tokens tlt-generate command line option --max-trials tlt-quantize command line option --max_new_tokens tlt-generate command line option --max_trials tlt-quantize command line option --model-dir tlt-benchmark command line option tlt-eval command line option tlt-generate command line option tlt-optimize command line option tlt-quantize command line option --model-name tlt-eval command line option tlt-generate command line option tlt-train command line option --model_dir tlt-benchmark command line option tlt-eval command line option tlt-generate command line option tlt-optimize command line option tlt-quantize command line option --model_name tlt-eval command line option tlt-generate command line option tlt-train command line option --nnodes tlt-train command line option --nproc-per-node tlt-train command line option --nproc_per_node tlt-train command line option --num-beams tlt-generate command line option --num_beams tlt-generate command line option --output-dir tlt-benchmark command line option tlt-optimize command line option tlt-quantize command line option tlt-train command line option --output_dir tlt-benchmark command line option tlt-optimize command line option tlt-quantize command line option tlt-train command line option --prompt tlt-generate command line option --prompt-with-context tlt-eval command line option tlt-train command line option --prompt-without-context tlt-eval command line option tlt-train command line option --prompt_with_context tlt-eval command line option tlt-train command line option --prompt_without_context tlt-eval command line option tlt-train command line option --repetition-penalty tlt-generate command line option --repetition_penalty tlt-generate command line option --response-key tlt-train command line option --response_key tlt-train command line option --simsiam tlt-train command line option --temperature tlt-generate command line option --timeout tlt-quantize command line option --top-k tlt-generate command line option --top-p tlt-generate command line option --top_k tlt-generate command line option --top_p tlt-generate command line option --use-case tlt-list-models command line option tlt-train command line option --use-horovod tlt-train command line option --use_case tlt-list-models command line option tlt-train command line option --use_horovod tlt-train command line option --verbose tlt-list-models command line option -d tlt-train command line option -f tlt-list-models command line option tlt-train command line option

_

__init__() (tlt.datasets.dataset.BaseDataset method) (tlt.datasets.hf_dataset.HFDataset method) (tlt.datasets.image_anomaly_detection.pytorch_custom_image_anomaly_detection_dataset.PyTorchCustomImageAnomalyDetectionDataset method) (tlt.datasets.image_classification.image_classification_dataset.ImageClassificationDataset method) (tlt.datasets.image_classification.pytorch_custom_image_classification_dataset.PyTorchCustomImageClassificationDataset method) (tlt.datasets.image_classification.tf_custom_image_classification_dataset.TFCustomImageClassificationDataset method) (tlt.datasets.image_classification.tfds_image_classification_dataset.TFDSImageClassificationDataset method) (tlt.datasets.image_classification.torchvision_image_classification_dataset.TorchvisionImageClassificationDataset method) (tlt.datasets.pytorch_dataset.PyTorchDataset method) (tlt.datasets.text_classification.hf_custom_text_classification_dataset.HFCustomTextClassificationDataset method) (tlt.datasets.text_classification.hf_text_classification_dataset.HFTextClassificationDataset method) (tlt.datasets.text_classification.text_classification_dataset.TextClassificationDataset method) (tlt.datasets.text_classification.tf_custom_text_classification_dataset.TFCustomTextClassificationDataset method) (tlt.datasets.text_classification.tfds_text_classification_dataset.TFDSTextClassificationDataset method) (tlt.datasets.text_generation.hf_custom_text_generation_dataset.HFCustomTextGenerationDataset method) (tlt.datasets.text_generation.text_generation_dataset.TextGenerationDataset method) (tlt.datasets.tf_dataset.TFDataset method) (tlt.models.hf_model.HFModel method) (tlt.models.image_anomaly_detection.pytorch_image_anomaly_detection_model.PyTorchImageAnomalyDetectionModel method) (tlt.models.image_anomaly_detection.torchvision_image_anomaly_detection_model.TorchvisionImageAnomalyDetectionModel method) (tlt.models.image_classification.image_classification_model.ImageClassificationModel method) (tlt.models.image_classification.keras_image_classification_model.KerasImageClassificationModel method) (tlt.models.image_classification.pytorch_hub_image_classification_model.PyTorchHubImageClassificationModel method) (tlt.models.image_classification.pytorch_image_classification_model.PyTorchImageClassificationModel method) (tlt.models.image_classification.tf_image_classification_model.TFImageClassificationModel method) (tlt.models.image_classification.tfhub_image_classification_model.TFHubImageClassificationModel method) (tlt.models.image_classification.torchvision_image_classification_model.TorchvisionImageClassificationModel method) (tlt.models.model.BaseModel method) (tlt.models.pytorch_model.PyTorchModel method) (tlt.models.text_classification.pytorch_hf_text_classification_model.PyTorchHFTextClassificationModel method) (tlt.models.text_classification.text_classification_model.TextClassificationModel method) (tlt.models.text_classification.tf_hf_text_classification_model.TFHFTextClassificationModel method) (tlt.models.text_classification.tf_text_classification_model.TFTextClassificationModel method) (tlt.models.text_generation.pytorch_hf_text_generation_model.PyTorchHFTextGenerationModel method) (tlt.models.text_generation.text_generation_model.TextGenerationModel method) (tlt.models.tf_model.TFModel method)

B

BaseDataset (class in tlt.datasets.dataset) BaseModel (class in tlt.models.model) benchmark() (tlt.models.image_anomaly_detection.pytorch_image_anomaly_detection_model.PyTorchImageAnomalyDetectionModel method) (tlt.models.image_anomaly_detection.torchvision_image_anomaly_detection_model.TorchvisionImageAnomalyDetectionModel method) (tlt.models.image_classification.image_classification_model.ImageClassificationModel method) (tlt.models.image_classification.keras_image_classification_model.KerasImageClassificationModel method) (tlt.models.image_classification.pytorch_hub_image_classification_model.PyTorchHubImageClassificationModel method) (tlt.models.image_classification.pytorch_image_classification_model.PyTorchImageClassificationModel method) (tlt.models.image_classification.tf_image_classification_model.TFImageClassificationModel method) (tlt.models.image_classification.tfhub_image_classification_model.TFHubImageClassificationModel method) (tlt.models.image_classification.torchvision_image_classification_model.TorchvisionImageClassificationModel method) (tlt.models.text_classification.pytorch_hf_text_classification_model.PyTorchHFTextClassificationModel method) (tlt.models.text_classification.text_classification_model.TextClassificationModel method) (tlt.models.text_classification.tf_hf_text_classification_model.TFHFTextClassificationModel method) (tlt.models.text_classification.tf_text_classification_model.TFTextClassificationModel method)

G

get_dataset() (in module tlt.datasets.dataset_factory)

get_model() (in module tlt.models.model_factory) get_supported_models() (in module tlt.models.model_factory)

H

HFCustomTextClassificationDataset (class in tlt.datasets.text_classification.hf_custom_text_classification_dataset) HFCustomTextGenerationDataset (class in tlt.datasets.text_generation.hf_custom_text_generation_dataset)

HFDataset (class in tlt.datasets.hf_dataset) HFModel (class in tlt.models.hf_model) HFTextClassificationDataset (class in tlt.datasets.text_classification.hf_text_classification_dataset)

I

ImageClassificationDataset (class in tlt.datasets.image_classification.image_classification_dataset)

ImageClassificationModel (class in tlt.models.image_classification.image_classification_model)

K

KerasImageClassificationModel (class in tlt.models.image_classification.keras_image_classification_model)

L

load_dataset() (in module tlt.datasets.dataset_factory)

load_model() (in module tlt.models.model_factory)

M

module tlt.datasets.dataset_factory tlt.models.model_factory

O

optimize_graph() (tlt.models.image_classification.keras_image_classification_model.KerasImageClassificationModel method) (tlt.models.image_classification.tf_image_classification_model.TFImageClassificationModel method) (tlt.models.image_classification.tfhub_image_classification_model.TFHubImageClassificationModel method) (tlt.models.text_classification.tf_hf_text_classification_model.TFHFTextClassificationModel method) (tlt.models.text_classification.tf_text_classification_model.TFTextClassificationModel method)

P

print_supported_models() (in module tlt.models.model_factory) PyTorchCustomImageAnomalyDetectionDataset (class in tlt.datasets.image_anomaly_detection.pytorch_custom_image_anomaly_detection_dataset) PyTorchCustomImageClassificationDataset (class in tlt.datasets.image_classification.pytorch_custom_image_classification_dataset) PyTorchDataset (class in tlt.datasets.pytorch_dataset) PyTorchHFTextClassificationModel (class in tlt.models.text_classification.pytorch_hf_text_classification_model)

PyTorchHFTextGenerationModel (class in tlt.models.text_generation.pytorch_hf_text_generation_model) PyTorchHubImageClassificationModel (class in tlt.models.image_classification.pytorch_hub_image_classification_model) PyTorchImageAnomalyDetectionModel (class in tlt.models.image_anomaly_detection.pytorch_image_anomaly_detection_model) PyTorchImageClassificationModel (class in tlt.models.image_classification.pytorch_image_classification_model) PyTorchModel (class in tlt.models.pytorch_model)

Q

quantize() (tlt.models.image_anomaly_detection.pytorch_image_anomaly_detection_model.PyTorchImageAnomalyDetectionModel method) (tlt.models.image_anomaly_detection.torchvision_image_anomaly_detection_model.TorchvisionImageAnomalyDetectionModel method) (tlt.models.image_classification.image_classification_model.ImageClassificationModel method) (tlt.models.image_classification.keras_image_classification_model.KerasImageClassificationModel method) (tlt.models.image_classification.pytorch_hub_image_classification_model.PyTorchHubImageClassificationModel method) (tlt.models.image_classification.pytorch_image_classification_model.PyTorchImageClassificationModel method) (tlt.models.image_classification.tf_image_classification_model.TFImageClassificationModel method) (tlt.models.image_classification.tfhub_image_classification_model.TFHubImageClassificationModel method) (tlt.models.image_classification.torchvision_image_classification_model.TorchvisionImageClassificationModel method) (tlt.models.text_classification.pytorch_hf_text_classification_model.PyTorchHFTextClassificationModel method) (tlt.models.text_classification.text_classification_model.TextClassificationModel method) (tlt.models.text_classification.tf_hf_text_classification_model.TFHFTextClassificationModel method) (tlt.models.text_classification.tf_text_classification_model.TFTextClassificationModel method)

T

TextClassificationDataset (class in tlt.datasets.text_classification.text_classification_dataset) TextClassificationModel (class in tlt.models.text_classification.text_classification_model) TextGenerationDataset (class in tlt.datasets.text_generation.text_generation_dataset) TextGenerationModel (class in tlt.models.text_generation.text_generation_model) TFCustomImageClassificationDataset (class in tlt.datasets.image_classification.tf_custom_image_classification_dataset) TFCustomTextClassificationDataset (class in tlt.datasets.text_classification.tf_custom_text_classification_dataset) TFDataset (class in tlt.datasets.tf_dataset) TFDSImageClassificationDataset (class in tlt.datasets.image_classification.tfds_image_classification_dataset) TFDSTextClassificationDataset (class in tlt.datasets.text_classification.tfds_text_classification_dataset) TFHFTextClassificationModel (class in tlt.models.text_classification.tf_hf_text_classification_model) TFHubImageClassificationModel (class in tlt.models.image_classification.tfhub_image_classification_model) TFImageClassificationModel (class in tlt.models.image_classification.tf_image_classification_model) TFModel (class in tlt.models.tf_model) TFTextClassificationModel (class in tlt.models.text_classification.tf_text_classification_model) tlt-benchmark command line option --batch-size --batch_size --dataset-dir --dataset-file --dataset_dir --dataset_file --delimiter --model-dir --model_dir --output-dir --output_dir tlt-eval command line option --class-names --class_names --dataset-catalog --dataset-dir --dataset-file, [1] --dataset-name --dataset_catalog --dataset_dir --dataset_file, [1] --dataset_name --delimiter --model-dir --model-name --model_dir --model_name --prompt-with-context --prompt-without-context --prompt_with_context --prompt_without_context tlt-generate command line option --max-new-tokens --max_new_tokens --model-dir --model-name --model_dir --model_name --num-beams --num_beams --prompt --repetition-penalty --repetition_penalty --temperature --top-k --top-p --top_k --top_p tlt-list-models command line option --framework --use-case --use_case --verbose -f tlt-optimize command line option --model-dir --model_dir --output-dir --output_dir tlt-quantize command line option --accuracy-criterion --accuracy_criterion --approach --batch-size --batch_size --dataset-dir --dataset-file --dataset_dir --dataset_file --delimiter --max-trials --max_trials --model-dir --model_dir --output-dir --output_dir --timeout

tlt-train command line option --add-aug --add_aug --class-names --class_names --context-key --context_key --cutpaste --dataset-catalog --dataset-dir --dataset-file --dataset-name --dataset_catalog --dataset_dir --dataset_file --dataset_name --delimiter --distributed --early-stopping --early_stopping --epochs --framework --hostfile --hvd-start-timeout --hvd_start_timeout --init-checkpoints --init_checkpoints --instruction-key --instruction_key --ipex-optimize --ipex_optimize --lora-alpha --lora-dropout --lora-rank --lora_alpha --lora_dropout --lora_rank --lr-decay --lr_decay --model-name --model_name --nnodes --nproc-per-node --nproc_per_node --output-dir --output_dir --prompt-with-context --prompt-without-context --prompt_with_context --prompt_without_context --response-key --response_key --simsiam --use-case --use-horovod --use_case --use_horovod -d -f tlt.datasets.dataset_factory module tlt.models.model_factory module TorchvisionImageAnomalyDetectionModel (class in tlt.models.image_anomaly_detection.torchvision_image_anomaly_detection_model) TorchvisionImageClassificationDataset (class in tlt.datasets.image_classification.torchvision_image_classification_dataset) TorchvisionImageClassificationModel (class in tlt.models.image_classification.torchvision_image_classification_model) train() (tlt.models.image_anomaly_detection.pytorch_image_anomaly_detection_model.PyTorchImageAnomalyDetectionModel method) (tlt.models.image_anomaly_detection.torchvision_image_anomaly_detection_model.TorchvisionImageAnomalyDetectionModel method) (tlt.models.image_classification.image_classification_model.ImageClassificationModel method) (tlt.models.image_classification.keras_image_classification_model.KerasImageClassificationModel method) (tlt.models.image_classification.pytorch_hub_image_classification_model.PyTorchHubImageClassificationModel method) (tlt.models.image_classification.pytorch_image_classification_model.PyTorchImageClassificationModel method) (tlt.models.image_classification.tf_image_classification_model.TFImageClassificationModel method) (tlt.models.image_classification.tfhub_image_classification_model.TFHubImageClassificationModel method) (tlt.models.image_classification.torchvision_image_classification_model.TorchvisionImageClassificationModel method) (tlt.models.text_classification.pytorch_hf_text_classification_model.PyTorchHFTextClassificationModel method) (tlt.models.text_classification.text_classification_model.TextClassificationModel method) (tlt.models.text_classification.tf_hf_text_classification_model.TFHFTextClassificationModel method) (tlt.models.text_classification.tf_text_classification_model.TFTextClassificationModel method) (tlt.models.text_generation.pytorch_hf_text_generation_model.PyTorchHFTextGenerationModel method) (tlt.models.text_generation.text_generation_model.TextGenerationModel method)