torch\_concepts.data.ToyDAGDataModule
=====================================

.. currentmodule:: torch_concepts.data

.. autoclass:: ToyDAGDataModule

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

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