
    8jS                       % S SK Jr  S SKrS SKJr  S SKJr  S SKJrJ	r	  \" \
5      R                  5       R                  S   r\S-  S-  S	-  r\S-  S-  S
-  rSS jr\" 5       rS\S'   \" \5       V Vs0 s H  u  pX_M	     snn rS\S'   \R+                  5        VV s0 s H  u  pX_M	     sn nrS\S'   \" \5      r\S:X  d   e\" SS9 " S S5      5       rgs  snn f s  sn nf )    )annotationsN)	dataclass)Path)DictTuple   dataemotionzemotion_labels.jsonzemotion_vad_anchors.jsonTuple[str, ...]c                    [         R                  " [        R                  SS95      S   n [	        U S S9n U  Vs/ s H  oS   PM	     nnU[        [        S5      5      :X  d
   SU 35       e[        S	 U  5       5      $ s  snf )
Nzutf-8)encodinglabelsc                    U S   $ )Nid )xs    ]/home/ubuntu/service/kemix-engine/package/face/animasync-face-v3/models/microalbert/config.py<lambda>_load_labels.<locals>.<lambda>   s    ag    )keyr      z+emotion_labels.json ids must be 0..15, got c              3  *   #    U  H	  oS    v   M     g7f)nameNr   ).0r   s     r   	<genexpr>_load_labels.<locals>.<genexpr>   s     )Dq6Ds   )jsonloadsLABELS_PATH	read_textsortedlistrangetuple)r	   r   idss      r   _load_labelsr'      s    ::k++W+=>xHD$-.D 
!DqT7DC
!$uRy/!V%PQTPU#VV!)D))) "s   A>EMOTION_LABELSzDict[str, int]EMOTION_TO_IDzDict[int, str]ID_TO_EMOTIONr   T)frozenc                     \ rS rSr% SrS\S'   SrS\S'   SrS\S'   S	rS\S
'   Sr	S\S'   Sr
S\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S '   S!rS\S"'   S#rS\S$'   S%rS&\S''   SrS\S('   S)rS\S*'   S+rS\S,'   S-rS\S.'   SrS\S/'   S0rS\S1'   S2rS\S3'   S4rS\S5'   S6r S7\S8'   S9r!S\S:'   S;r"S\S<'   Sr#S\S='   S>r$S7\S?'   S@r%S\SA'   SBr&S\SC'   S4r'S\SD'   SEr(S\SF'   SGr)gH)IMicroAlbertConfig   i>  int
vocab_sizer   pad_token_id   embedding_sizei  hidden_size   
num_layers	num_headsi   ffn_sizemax_seq_leng?floatdropoutg333333?attention_dropoutg-q=layer_norm_epsr   num_emotions   vad_dim@   vad_head_hiddeng{Gz?initializer_rangelabel_smoothingg      ?vad_loss_weight_max   vad_warmup_epochs)      ?rH         ?zTuple[float, float, float]vad_dim_weightssnap_loss_weightr   snap_start_epochg      ?snap_conf_thresholdg       @snap_softmax_temp
snap_levelg?range_loss_weightgffffff?range_target_stdg?smooth_l1_beta)rI   g      @zTuple[float, float]class_weight_clipg-C6*?backbone_lrgMbP?head_lrweight_decay)g?g\(\?
adam_betasgư>adam_epsrH   	grad_clipwarmup_ratio*   seedr   N)*__name__
__module____qualname____firstlineno__r0   __annotations__r1   r3   r4   r6   r7   r8   r9   r;   r<   r=   r>   r@   rB   rC   rD   rE   rG   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r\   __static_attributes__r   r   r   r-   r-      sw   JL#NCKJIsHcKGU#u#!NE!L#GSOS#u#!OU!!$$s2AO/A!e!c!%%"u"J#u#"e"NE-7*7KGUL%&1J#1HeIuL%D#Nr   r-   )returnr   )
__future__r   r   dataclassesr   pathlibr   typingr   r   __file__resolveparents	REPO_ROOTr    ANCHORS_PATHr'   r(   ra   	enumerater)   itemsr*   lenNUM_EMOTIONSr-   )ir   s   00r   <module>rr      s    "  !  N""$,,Q/	& 9,/DD6!I-0JJ* #/. 08A.8Q R8QWQ8Q R~ R8E8K8K8M N8MWT8M N~ N>"r  $' ' ' !S Ns   5CC 