o Ti@sddlmZddlmZddlmZddZGdddeZGdd d eZGd d d eZ Gd d d eZ GdddeZ dS))Optional)model_validator)DeepSpeedConfigModelcs fdddD}tdi|S)Ncsi|] }||iqS)get).0key param_dictrL/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/monitor/config.py sz&get_monitor_config..) tensorboardwandb csv_monitorcometr)DeepSpeedMonitorConfig)r monitor_dictrr r get_monitor_config src@:eZdZUdZdZeed< dZeed< dZ eed<dS) TensorBoardConfigz(Sets parameters for TensorBoard monitor.Fenabled output_pathDeepSpeedJobNamejob_nameN __name__ __module__ __qualname____doc__rbool__annotations__rstrrrrrr r    rc@sPeZdZUdZdZeed< dZee ed< dZ ee ed< dZ e ed<dS) WandbConfigz"Sets parameters for WandB monitor.FrNgroupteam deepspeedproject) rrrrrr r!r%rr"r&r(rrrr r$!s   r$c@r) CSVConfigz Sets parameters for CSV monitor.FrrrrrNrrrrr r)1r#r)c@seZdZUdZdZeed< dZeed< dZ e e ed< dZ e e ed< dZ e e ed < dZe e ed < dZe e ed < dZe eed < dZe e ed <dS) CometConfigz Sets parameters for Comet monitor. For logging data Comet uses experiment object. https://www.comet.com/docs/v2/api-and-sdk/python-sdk/reference/Experiment/ Frdsamples_log_intervalNr( workspaceapi_keyexperiment_nameexperiment_keyonlinemode)rrrrrr r!r,intr(rr"r-r.r/r0r1r2rrrr r*As(   r*c@s\eZdZUdZiZeed< iZeed< iZ e ed< iZ e ed< e dddd Zd S) rz/Sets parameters for various monitoring methods.r rrrafter)r2cCs.|jjp|jjp|jjp|jj}||jd<|S)Nr)r rrrr__dict__)selfrrrr check_enableds  z$DeepSpeedMonitorConfig.check_enabledN)rrrrr rr!rr*rr$rr)rr7rrrr r}s     rN) typingrpydanticrdeepspeed.runtime.config_utilsrrrr$r)r*rrrrr s   <