torch\_concepts.data.base.datamodule.ConceptDataModule
======================================================

.. currentmodule:: torch_concepts.data.base.datamodule

.. autoclass:: ConceptDataModule

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

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