ertk.pytorch.dataset.MTLDataModule
==================================

.. currentmodule:: ertk.pytorch.dataset

.. autoclass:: MTLDataModule
    :members:
    :show-inheritance:

    
    .. automethod:: __init__

    
    .. rubric:: Methods

    .. autosummary::
    
        ~MTLDataModule.__init__

    
    .. rubric:: Inherited Methods
    .. autosummary::
    
        ~MTLDataModule.from_datasets
        ~MTLDataModule.load_from_checkpoint
        ~MTLDataModule.load_state_dict
        ~MTLDataModule.on_after_batch_transfer
        ~MTLDataModule.on_before_batch_transfer
        ~MTLDataModule.on_exception
        ~MTLDataModule.predict_dataloader
        ~MTLDataModule.prepare_data
        ~MTLDataModule.save_hyperparameters
        ~MTLDataModule.setup
        ~MTLDataModule.state_dict
        ~MTLDataModule.teardown
        ~MTLDataModule.test_dataloader
        ~MTLDataModule.train_dataloader
        ~MTLDataModule.transfer_batch_to_device
        ~MTLDataModule.val_dataloader
    
    
    

    
    
    .. rubric:: Attributes

    .. autosummary::
    
        ~MTLDataModule.CHECKPOINT_HYPER_PARAMS_KEY
        ~MTLDataModule.CHECKPOINT_HYPER_PARAMS_NAME
        ~MTLDataModule.CHECKPOINT_HYPER_PARAMS_TYPE
        ~MTLDataModule.hparams
        ~MTLDataModule.hparams_initial
        ~MTLDataModule.name
        ~MTLDataModule.train_dataset
        ~MTLDataModule.val_dataset
        ~MTLDataModule.test_dataset