training/PRETRAIN/__init__.py

"""
Domain Adaptation Pretraining Module for LKAlert

This module provides tools for adapting VLM backbones (Qwen2.5-VL) to
autonomous driving tasks through multi-task pretraining.

Components:
- DomainAdaptationConfig: Configuration classes
- DrivingDomainDataset: Multi-task dataset for driving scenes
- DomainAdaptationTrainer: LoRA-based trainer

Usage:
    from training.pretrain import (
        DomainAdaptationConfig,
        DrivingDomainDataset,
        DomainAdaptationTrainer,
        get_7b_config,
    )
"""

from .domain_adaptation_config import (
    DomainAdaptationConfig,
    ModelConfig,
    DataConfig,
    TrainingConfig,
    get_default_config,
    get_quick_test_config,
    get_7b_config,
    get_3b_config,
)

from .domain_adaptation_dataset import (
    DrivingDomainDataset,
    collate_fn,
    create_dataloaders,
)

from .domain_adaptation_trainer import DomainAdaptationTrainer

__all__ = [
    # Configs
    "DomainAdaptationConfig",
    "ModelConfig",
    "DataConfig",
    "TrainingConfig",
    "get_default_config",
    "get_quick_test_config",
    "get_7b_config",
    "get_3b_config",
    # Dataset
    "DrivingDomainDataset",
    "collate_fn",
    "create_dataloaders",
    # Trainer
    "DomainAdaptationTrainer",
]