o
    ´‹
j<  ã                   @   s   G d d„ dƒZ dS )c                   @   sn   e Zd Zeddd„ƒZeddd„ƒZeddd„ƒZedd	d
„ƒZeddd„ƒZeddd„ƒZ	eddd„ƒZ
dS )ÚTrainerCallbackÚreturnNc                 C   óz   t | jdƒrt | jjdƒr| jj | ¡ nt | jdƒr!| j | ¡ t | jdƒr-| j | ¡ t | jdƒr;| j | ¡ d S d S )NÚmoduleÚon_init_start)ÚhasattrÚmodelr   r   Ú	criterionÚ	optimizer©Útrainer© r   úF/home/kuhnn/.local/lib/python3.10/site-packages/TTS/utils/callbacks.pyr      ó   €ÿzTrainerCallback.on_init_startc                 C   r   )Nr   Úon_init_end)r   r   r   r   r   r	   r
   r   r   r   r      r   zTrainerCallback.on_init_endc                 C   r   )Nr   Úon_epoch_start)r   r   r   r   r   r	   r
   r   r   r   r       r   zTrainerCallback.on_epoch_startc                 C   r   )Nr   Úon_epoch_end)r   r   r   r   r   r	   r
   r   r   r   r   /   r   zTrainerCallback.on_epoch_endc                 C   r   )Nr   Úon_train_step_start)r   r   r   r   r   r	   r
   r   r   r   r   >   r   z#TrainerCallback.on_train_step_startc                 C   r   )Nr   Úon_train_step_end)r   r   r   r   r   r	   r
   r   r   r   r   M   r   z!TrainerCallback.on_train_step_endc                 C   r   )Nr   Úon_keyboard_interrupt)r   r   r   r   r   r	   r
   r   r   r   r   \   r   z%TrainerCallback.on_keyboard_interrupt)r   N)Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr   r   r   r   r   r   r   r   r   r   r   r      s    r   N)r   r   r   r   r   Ú<module>   s    