torch\_concepts.data.CUBDataModule
==================================

.. currentmodule:: torch_concepts.data

.. autoclass:: CUBDataModule

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

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