o iS@sUddlmZddlZddlmZddlmZddlmZm Z ee  j dZ e ddd Ze ddd Zdd dZeZd ed<ddeeDZded<ddeDZded<eeZedkshJeddGdddZdS)) annotationsN) dataclass)Path)DictTupledataemotionzemotion_labels.jsonzemotion_vad_anchors.jsonreturnTuple[str, ...]cCsdttjddd}t|ddd}dd|D}|ttd ks)Jd |td d |DS) Nzutf-8)encodinglabelscSs|dS)Nid)xrrQ/dataset/kemix-engine/package/face/animasync-face-v3/models/microalbert/config.pysz_load_labels..)keycSsg|]}|dqS)rr.0rrrr sz _load_labels..z+emotion_labels.json ids must be 0..15, got css|]}|dVqdS)nameNrrrrr sz_load_labels..)jsonloads LABELS_PATH read_textsortedlistrangetuple)ridsrrr _load_labels s r#EMOTION_LABELScCi|]\}}||qSrr)rirrrr r'zDict[str, int] EMOTION_TO_IDcCr%rr)rrr&rrrr'r(zDict[int, str] ID_TO_EMOTIONrT)frozenc@seZdZUdZded<dZded<dZded<dZded <d Zded <d Z ded <d Z ded<dZ ded<dZ ded<dZ ded<dZded<dZded<dZded<dZded<dZded<dZded<d Zded!<d"Zded#<d$Zd%ed&<dZded'<d(Zded)<d*Zded+<d,Zded-<dZded.<d/Zded0<d1Zded2<d3Zded4<d5Zd6ed7<d8Z ded9<d:Z!ded;<dZ"ded<<d=Z#d6ed><d?Z$ded@<dAZ%dedB<d3Z&dedC<dDZ'dedE<dFS)GMicroAlbertConfigi>int vocab_sizer pad_token_idembedding_sizei hidden_size num_layers num_headsiffn_size max_seq_leng?floatdropoutg333333?attention_dropoutg-q=layer_norm_epsr num_emotionsvad_dim@vad_head_hiddeng{Gz?initializer_rangelabel_smoothingg?vad_loss_weight_maxvad_warmup_epochs)?rF?zTuple[float, float, float]vad_dim_weightssnap_loss_weightrsnap_start_epochg?snap_conf_thresholdg@snap_softmax_temp snap_levelg?range_loss_weightgffffff?range_target_stdg?smooth_l1_beta)rGg@zTuple[float, float]class_weight_clipg-C6*? backbone_lrgMbP?head_lr weight_decay)g?g\(\? adam_betasgư>adam_epsrF grad_clip warmup_ratio*seedN)(__name__ __module__ __qualname__r.__annotations__r/r1r2r4r5r6r7r9r:r;r<r>r@rArBrCrErHrIrJrKrLrMrNrOrPrQrRrSrTrUrVrWrXrZrrrrr,sJ                                   r,)r r ) __future__rr dataclassesrpathlibrtypingrr__file__resolveparents REPO_ROOTr ANCHORS_PATHr#r$r^ enumerater)itemsr*len NUM_EMOTIONSr,rrrrs