
    RЦi                     L    S r SSKJr  SSKJr  SSKJrJrJr  SSK	J
r
Jr  / SQrg)zTraining task abstractions for timm.

This module provides task-based abstractions for training loops where each task
encapsulates both the forward pass and loss computation, returning a dictionary
with loss components and outputs for logging.
   )TrainingTask)ClassificationTask)DistillationTeacherLogitDistillationTaskFeatureDistillationTask)TokenDistillationTeacherTokenDistillationTask)r   r   r   r   r   r   r	   N)__doc__taskr   classificationr   distillationr   r   r   token_distillationr   r	   __all__     Q/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/timm/task/__init__.py<module>r      s"     . ] ] Or   