
    
3jM                     \    S r SSKJr  SSKJr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)resume_task_checkpointload_task_ema_checkpoint)ClassificationTask)DistillationTeacherLogitDistillationTaskFeatureDistillationTask)TokenDistillationTeacherTokenDistillationTask)	r   r   r   r   r   r   r	   r
   r   N)__doc__taskr   _helpersr   r   classificationr   distillationr   r   r	   token_distillationr
   r   __all__     L/home/wildlama/miniconda3/lib/python3.13/site-packages/timm/task/__init__.py<module>r      s%     F . ] ] O
r   