o
    *jg                     @   sn  d dl mZ d dlmZ erzddlmZmZ ddlmZ ddl	m
Z
 ddlmZ ddlmZ dd	lmZ dd
lmZmZ ddlmZ ddlmZmZmZmZ ddlmZmZ ddlmZm Z m!Z! ddl"m#Z# ddl$m%Z% ddl&m'Z' ddl(m)Z) d%S ddgg ddgdgdgdgddgddgg dg ddgdgd gd!gd"d#gd$Z*d d%l+Z+ee,e- d& e*e.i d'e+j/e,< d%S )(    )TYPE_CHECKING)LazyImportModule   )HOOKS
build_hook)EarlyStopHook)SparsityHook)EvaluationHook)Hook)IterTimerHook)TensorboardHookTextLoggerHook)LrSchedulerHook)ApexAMPOptimizerHookNoneOptimizerHookOptimizerHookTorchAMPOptimizerHook)Priorityget_priority)CheckpointHookLoadCheckpointHookBestCkptSaverHook)DDPHook)DeepspeedHook)MegatronHook)	SwiftHookr   r   )r   r   r   r   r	   r
   r   r   r   r   ZNoneLrSchedulerHookr   r   r   r   r   r   )builderZcheckpoint_hookcompressionevaluation_hookhookiter_timer_hookloggerlr_scheduler_hook	optimizer
checkpointzdistributed.ddp_hookzdistributed.deepspeed_hookzdistributed.megatron_hookzswift.swift_hookpriorityN__file__)Zmodule_specZextra_objects)0typingr   Zmodelscope.utils.import_utilsr   r   r   r   Zearly_stop_hookr   r   r   r   r	   r   r
   r    r   r!   r   r   r"   r   r#   r   r   r   r   r%   r   r   r$   r   r   r   Zdistributed.ddp_hookr   Zdistributed.deepspeed_hookr   Zdistributed.megatron_hookr   Zswift.swift_hookr   Z_import_structuresys__name__globals__spec__modules r-   r-   c/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/trainers/hooks/__init__.py<module>   sT   