tlt.models.image\_classification.torchvision\_image\_classification\_model.TorchvisionImageClassificationModel
==============================================================================================================

.. currentmodule:: tlt.models.image_classification.torchvision_image_classification_model

.. autoclass:: TorchvisionImageClassificationModel

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~TorchvisionImageClassificationModel.__init__
      ~TorchvisionImageClassificationModel.benchmark
      ~TorchvisionImageClassificationModel.cleanup_saved_objects_for_distributed
      ~TorchvisionImageClassificationModel.evaluate
      ~TorchvisionImageClassificationModel.export
      ~TorchvisionImageClassificationModel.export_for_distributed
      ~TorchvisionImageClassificationModel.freeze_layer
      ~TorchvisionImageClassificationModel.list_layers
      ~TorchvisionImageClassificationModel.load_from_directory
      ~TorchvisionImageClassificationModel.optimize_graph
      ~TorchvisionImageClassificationModel.predict
      ~TorchvisionImageClassificationModel.quantize
      ~TorchvisionImageClassificationModel.train
      ~TorchvisionImageClassificationModel.unfreeze_layer
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~TorchvisionImageClassificationModel.do_fine_tuning
      ~TorchvisionImageClassificationModel.dropout_layer_rate
      ~TorchvisionImageClassificationModel.framework
      ~TorchvisionImageClassificationModel.image_size
      ~TorchvisionImageClassificationModel.learning_rate
      ~TorchvisionImageClassificationModel.model_name
      ~TorchvisionImageClassificationModel.num_classes
      ~TorchvisionImageClassificationModel.preprocessor
      ~TorchvisionImageClassificationModel.use_case