deeplabcut.pose_estimation_pytorch.config.training

Training configuration classes for DeepLabCut pose estimation models.

Classes:

Name	Description
TrainSettingsConfig	Training settings configuration.

TrainSettingsConfig

Bases: DLCBaseConfig

Training settings configuration.

Attributes:

Name	Type	Description
batch_size	int	Training batch size
dataloader_workers	int	Number of data loader workers
dataloader_pin_memory	bool	Whether to pin memory in data loader
display_iters	int	Display interval for training progress
epochs	int	Number of training epochs
seed	int	Random seed for reproducibility
weight_init	WeightInitialization | None	Weight initialization configuration

Source code in deeplabcut/pose_estimation_pytorch/config/training.py

class TrainSettingsConfig(DLCBaseConfig):
    """Training settings configuration.

    Attributes:
        batch_size: Training batch size
        dataloader_workers: Number of data loader workers
        dataloader_pin_memory: Whether to pin memory in data loader
        display_iters: Display interval for training progress
        epochs: Number of training epochs
        seed: Random seed for reproducibility
        weight_init: Weight initialization configuration
    """

    batch_size: int = 8
    dataloader_workers: int = 0
    dataloader_pin_memory: bool = False
    display_iters: int = 500
    epochs: int = 200
    seed: int = 42
    # @TODO @deruyter92 2026-02-13: V0 pipeline uses None for default initialization
    # (with ImageNet weights). We should update this to explicit WeightInitialization.
    weight_init: WeightInitialization | None = None