o
    o°×i£  ã                   @   s²  d dl mZ d dlZd dlmZ d dlmZ eƒ  dZ	dZ
dZdZdZd	ZdZd
ZdZedkrWee	eeee
eeeed	ZeddeedZe d eee ¡Zdgdg dgd  dgd  dgd  Zedgeeeƒ  7 Ze ee¡Zejdd\ZZejdd e ¡ Ze ¡ Z e j!j" #d¡ ej$eeddd \Z%ej$eed!d"d \Z&e j$eed#d$d \Z'e (d e¡ e )d d%¡ e )d d&¡ e  )d d'¡ e *d(¡ e +d¡ e +d"¡ e  +d$¡ ej,j- .e% /¡ ¡ ej,j- .e& /¡ ¡ e j,j- .e' /¡ ¡ e 0¡  e 1e2d edƒ¡ e  3g d)¢¡ e4d*d+d,Z5ej6d5d-e% /¡ d.œe5¤Ž ej6d5d-e& /¡ d.œe5¤Ž e j6d5d-e' /¡ d.œe5¤Ž ej6d5d/d0ie5¤Ž ej7e%e&e'gd1d2 e 8d3¡ e 9d4¡ dS dS )6é    N)Úinstall)Úcosine_schedulergü©ñÒMbP?é   g-Cëâ6?gíµ ÷Æ°>é   éÿÿÿÿéd   é   Ú__main__)Úinitial_ep_per_cycleÚcycle_decayÚ	cycle_mulgê-™—q=gš™™™™™©?)Úniter_per_epÚepochsé   é   é   é    é   é@   é
   é`   )r   r   )Úfigsizeg      è?)Úright)ÚaxesgìQ¸…ëñ?zb-zLearning Rate)Úlabelzr-zWeight Decayzg-z
Batch Sizeg/n£¼R?g)\Âõ(¬?é€   ÚEpochs)r   r   r   r   r   r   r   r   é   g      ø?)ÚsizeÚwidthÚy)ÚaxisÚcolorsr!   Úxzlower right)ÚhandlesÚlocTzout/lrs.pdf© ):Úmatplotlib.pyplotÚpyplotÚpltÚnumpyÚnpÚicecreamr   Údf.lrr   ÚLRÚWARMUPÚLR_WARMUP_INITÚLR_MINÚCYCLE_DECAYÚ	CYCLE_LIMÚ	CYCLE_MULÚ
NUM_EPOCHSÚSTEPS_PER_EPOCHÚ__name__ÚlrsÚwdsÚlinspaceÚtÚbsÚlenÚrepeatÚsubplotsÚfigÚaxÚsubplots_adjustÚtwinxÚtwin1Útwin2Úspinesr   Úset_positionÚplotÚp1Úp2Úp3Úset_xlimÚset_ylimÚ
set_xlabelÚ
set_ylabelÚyaxisr   Ú	set_colorÚ	get_colorÚgridÚ
set_xticksÚrangeÚ
set_yticksÚdictÚtkwÚtick_paramsÚlegendÚset_tight_layoutÚsavefigr&   r&   r&   úQ/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/df/scripts/plot_lrs.pyÚ<module>   s|    
÷*




Å