o ॵik@sBddlmZddlmZddlmZddlmZGdddZdS) )wraps) TrainerStages)is_method_overridden)Priorityc @s6eZdZdZejejejejej ej ej ej ej ejejejejf ZejZddZddZddZdd Zd d Zd d ZddZddZddZddZddZ ddZ ddZddZ ddZ d d!Z d"d#Zed$d%Zed&d'Zed(d)Zed*d+Zed,d-Zed.d/Z d0d1Z!d2d3Z"d4d5Z#d6S)7Hookzo The Hook base class of any modelscope trainer. You can build your own hook inherited from this class. cCdS)zN Will be called at the end of the trainer's `__init__` method Nselftrainerr r R/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/trainers/hooks/hook.py after_initszHook.after_initcCr)z Will be called before trainer loop begins. Args: trainer: The trainer instance. Returns: None Nr r r r r before_run zHook.before_runcCr)z Will be called after trainer loop end. Args: trainer: The trainer instance. Returns: None Nr r r r r after_run*rzHook.after_runcCr)z Will be called before eval loop begins. Args: trainer: The trainer instance. Returns: None Nr r r r r before_val5rzHook.before_valcCr)z Will be called after eval loop end. Args: trainer: The trainer instance. Returns: None Nr r r r r after_val@rzHook.after_valcCr)z Will be called before every epoch begins. Args: trainer: The trainer instance. Returns: None Nr r r r r before_epochKrzHook.before_epochcCr)z Will be called after every epoch ends. Args: trainer: The trainer instance. Returns: None Nr r r r r after_epochVrzHook.after_epochcCr)z Will be called before every loop begins. Args: trainer: The trainer instance. Returns: None Nr r r r r before_iterazHook.before_itercCr)z Will be called after every loop ends. Args: trainer: The trainer instance. Returns: None Nr r r r r after_iterkrzHook.after_itercC||dS)z Will be called before every train epoch begins. Default call ``self.before_epoch`` Args: trainer: The trainer instance. Returns: None Nrr r r r before_train_epochu zHook.before_train_epochcCr)z Will be called before every validation epoch begins. Default call ``self.before_epoch`` Args: trainer: The trainer instance. Returns: None Nrr r r r before_val_epochrzHook.before_val_epochcCr)z Will be called after every train epoch ends. Default call ``self.after_epoch`` Args: trainer: The trainer instance. Returns: None Nrr r r r after_train_epochrzHook.after_train_epochcCr)z Will be called after every validation epoch ends. Default call ``self.after_epoch`` Args: trainer: The trainer instance. Returns: None Nrr r r r after_val_epochrzHook.after_val_epochcCr)z Will be called before every train loop begins. Default call ``self.before_iter`` Args: trainer: The trainer instance. Returns: None Nrr r r r before_train_iterzHook.before_train_itercCr)z Will be called before every validation loop begins. Default call ``self.before_iter`` Args: trainer: The trainer instance. Returns: None Nr!r r r r before_val_iterr#zHook.before_val_itercCr)z Will be called after every train loop ends. Default call ``self.after_iter`` Args: trainer: The trainer instance. Returns: None Nrr r r r after_train_iterr#zHook.after_train_itercCr)z Will be called after every validation loop ends. Default call ``self.after_iter`` Args: trainer: The trainer instance. Returns: None Nr%r r r r after_val_iterr#zHook.after_val_itercC|dkr |jd|dkSdS)zK Whether to reach every ``n`` epochs Returns: bool rrF)epochr nr r r every_n_epochszHook.every_n_epochscCr()z^ Whether to reach every ``n`` iterations at every epoch Returns: bool rrF) inner_iter)runnerr+r r r every_n_inner_itersr-zHook.every_n_inner_iterscCr()zO Whether to reach every ``n`` iterations Returns: bool rrF)iterr*r r r every_n_itersr-zHook.every_n_iterscC|jd|jkS)zO Whether to reach the end of every epoch Returns: bool r)r.iters_per_epochr r r r end_of_epochzHook.end_of_epochcCr3)zG Whether to reach the last epoch Returns: bool r)r) max_epochsr5r r r is_last_epochr7zHook.is_last_epochcCr3)zj Whether to reach the last iteration in the entire training process Returns: bool r)r1 max_itersr5r r r is_last_iterr7zHook.is_last_itercs<ttjD] }t|t|r|qfddtjDS)Ncsg|]}|vr|qSr r ).0stagetrigger_stagesr r sz-Hook.get_triggered_stages..)setrstagesradd)r r=r r>r get_triggered_stagess    zHook.get_triggered_stagescCsiSNr )r r r r state_dictzHook.state_dictcCsdSrEr )r rFr r r load_state_dictrGzHook.load_state_dictN)$__name__ __module__ __qualname____doc__rrrrrr"r&rrr$r'r rrrBrNORMALPRIORITYrrrr staticmethodr,r0r2r6r9r;rDrFrHr r r r r sV                      rN) functoolsrmodelscope.utils.constantrmodelscope.utils.import_utilsrpriorityrrr r r r s