ertk.pytorch.train.PyTorchTrainConfig
=====================================

.. currentmodule:: ertk.pytorch.train

.. autoclass:: PyTorchTrainConfig
    :members:
    :show-inheritance:

    
    .. automethod:: __init__

    
    .. rubric:: Methods

    .. autosummary::
    
        ~PyTorchTrainConfig.__init__

    
    .. rubric:: Inherited Methods
    .. autosummary::
    
        ~PyTorchTrainConfig.default
        ~PyTorchTrainConfig.from_config
        ~PyTorchTrainConfig.from_file
        ~PyTorchTrainConfig.merge_with_args
        ~PyTorchTrainConfig.merge_with_config
        ~PyTorchTrainConfig.to_dictconfig
        ~PyTorchTrainConfig.to_file
        ~PyTorchTrainConfig.to_string
    
    
    

    
    
    .. rubric:: Attributes

    .. autosummary::
    
        ~PyTorchTrainConfig.batch_size
        ~PyTorchTrainConfig.data_config
        ~PyTorchTrainConfig.dist_strategy
        ~PyTorchTrainConfig.enable_checkpointing
        ~PyTorchTrainConfig.epochs
        ~PyTorchTrainConfig.logging
        ~PyTorchTrainConfig.n_gpus
        ~PyTorchTrainConfig.resume_checkpoint
        ~PyTorchTrainConfig.save_every_n_epochs
        ~PyTorchTrainConfig.wrapper_model_config