o
    ´©iÓ  ã                   @  s.   d dl mZ d dlmZ eG dd„ dƒƒZdS )é    )Úannotations)Ú	dataclassc                   @  sJ   e Zd ZU dZded< ded< ded< ded< ded< ded< ded	< d
S )ÚTrainingStatez¼
    Container that mirrors the object returned in the minicpm-audio training
    loop. It holds persistent references to the model, optimizer, scheduler,
    dataloaders and tracker.
    ÚobjectÚ	generatorÚ	optimizerÚ	schedulerÚtrain_loaderÚ
val_loaderÚtrackerÚbatch_processorN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__Ú__annotations__© r   r   úI/home/ubuntu/.local/lib/python3.10/site-packages/voxcpm/training/state.pyr      s   
 r   N)Ú
__future__r   Údataclassesr   r   r   r   r   r   Ú<module>   s    