torch\_concepts.data.AWA2DataModule
===================================

.. currentmodule:: torch_concepts.data

.. autoclass:: AWA2DataModule

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~AWA2DataModule.__init__
      ~AWA2DataModule.from_datasets
      ~AWA2DataModule.get_dataloader
      ~AWA2DataModule.load_from_checkpoint
      ~AWA2DataModule.load_state_dict
      ~AWA2DataModule.on_after_batch_transfer
      ~AWA2DataModule.on_before_batch_transfer
      ~AWA2DataModule.on_exception
      ~AWA2DataModule.predict_dataloader
      ~AWA2DataModule.prepare_data
      ~AWA2DataModule.remove_ignored_hparams
      ~AWA2DataModule.save_hyperparameters
      ~AWA2DataModule.setup
      ~AWA2DataModule.state_dict
      ~AWA2DataModule.teardown
      ~AWA2DataModule.test_dataloader
      ~AWA2DataModule.train_dataloader
      ~AWA2DataModule.transfer_batch_to_device
      ~AWA2DataModule.val_dataloader
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~AWA2DataModule.CHECKPOINT_HYPER_PARAMS_KEY
      ~AWA2DataModule.CHECKPOINT_HYPER_PARAMS_NAME
      ~AWA2DataModule.CHECKPOINT_HYPER_PARAMS_TYPE
      ~AWA2DataModule.backbone
      ~AWA2DataModule.hparams
      ~AWA2DataModule.hparams_initial
      ~AWA2DataModule.n_samples
      ~AWA2DataModule.name
      ~AWA2DataModule.test_len
      ~AWA2DataModule.testset
      ~AWA2DataModule.train_len
      ~AWA2DataModule.trainset
      ~AWA2DataModule.val_len
      ~AWA2DataModule.valset