
    9if                     F    S SK r S SKJr  SSKJr  SSKJr   " S S\5      rg)	    N)Path   )BaseTrainer   )MODELSc                   >    \ rS rSrSr\rS\4S jrS r	S\
4S jrSrg	)
VideoClsTrainer   z"Image Classification Model Trainersrc_label_dict_pathc                     [        U R                  R                  5      R                  S5      n[        R
                  " X5        g)zbdump label dict config

Args:
    src_label_dict_path (str): path to label dict file to be saved.
zlabel_dict.txtN)r   global_configoutputjoinpathshutilcopyfile)selfr   dst_label_dict_paths      l/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/paddlex/modules/video_classification/trainer.pydump_label_dictVideoClsTrainer.dump_label_dict   s4     #4#5#5#<#<=FFGWX+A    c                 ,   U R                   R                  (       a/  U R                  R                  U R                   R                  5        U R                   R                  (       a/  U R                  R                  U R                   R                  5        U R                   R                  (       a/  U R                  R                  U R                   R                  5        U R                  R                  U R                  R                  S5        U R                   R                  b/  U R                  R                  U R                   R                  5        U R                   R                  S:w  a/  U R                  R                  U R                   R                  5        [        U R                  R                  5      R!                  S5      nUR#                  5       (       a  U R%                  U5        U R                   R&                  b.  U R                  R)                  U R                   R&                  SS9  U R*                  R&                  b.  U R                  R)                  U R*                  R&                  SS9  U R                   R,                  b/  U R                  R/                  U R                   R,                  5        U R                   R0                  b/  U R                  R3                  U R                   R0                  5        U R                   R4                  b/  U R                  R7                  U R                   R4                  5        U R                  R8                  b0  U R                  R;                  U R                  R8                  5        gg)zupdate training configVideoClsDatasetN z	label.txttrain)modeeval)train_configlog_interval
pdx_configupdate_log_intervaleval_intervalupdate_eval_intervalsave_intervalupdate_save_intervalupdate_datasetr   dataset_dirnum_classesupdate_num_classespretrain_weight_pathupdate_pretrained_weightsr   r   existsr   
batch_sizeupdate_batch_sizeeval_configlearning_rateupdate_learning_rateepochs_iters_update_epochswarmup_stepsupdate_warmup_epochsr   _update_output_dir)r   label_dict_paths     r   update_configVideoClsTrainer.update_config$   s   ))OO//0A0A0N0NO**OO001B1B1P1PQ**OO001B1B1P1PQ&&**,=	
 ((4OO..t/@/@/L/LM11R7OO55!!66 t11==>GGT!!##  1''3OO--!!,,7 .  &&2OO--d.>.>.I.IPV-W**6OO001B1B1P1PQ))5OO**4+<+<+I+IJ))5OO001B1B1O1OP$$0OO..t/A/A/H/HI 1r   returnc                 :   SU R                  5       0nU R                  R                  b3  U R                  R                  S:w  a  U R                  R                  US'   U R                  R                  SS5      US'   U R                  R                  SS5      US'   U$ )zkget key-value arguments of model training function

Returns:
    dict: the arguments of training function.
devicer   resume_pathdy2stFampOFF)
get_devicer   r=   get)r   
train_argss     r   get_train_kwargs VideoClsTrainer.get_train_kwargsI   s      12
))5!!--3(,(9(9(E(EJ}%"//33GUC
7 --11%?
5r    N)__name__
__module____qualname____firstlineno____doc__r   entitiesstrr   r8   dictrD   __static_attributes__rF   r   r   r	   r	      s+    ,HB3 B#JJ$ r   r	   )r   pathlibr   baser   
model_listr   r	   rF   r   r   <module>rS      s!       Bk Br   