o 1j;@s<ddlZddlZddlmZddlmZGdddeZdS)Nabspath) BaseConfigc@seZdZddZddZddZddZd'd d Zd'd d Zd(ddZ ddZ ddZ d'ddZ ddZ ddZddZddZdd Zd!d"Zd#efd$d%Zd&S)) PP3DConfigcs&fdd||j}||dS)NcsD|D]\}}t|tr||vr|||||<q|||<q|SN)items isinstancedict)Z dict_fromZdict_tokeyval_merge_config_dictsk/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/repo_apis/Paddle3D_api/pp3d_config.pyrs  z.PP3DConfig.update.._merge_config_dicts)r reset_from_dict)selfZ dict_like_objdict_rrrupdates  zPP3DConfig.updatecCsPt|dd}tj|tjd}Wdn1swY|}||dS)Nrutf-8)Loader)codecsopenyamlloadZ FullLoaderr)r config_pathfileZdicrrrrr's zPP3DConfig.loadcCsBt|ddd}t|j|WddS1swYdS)Nwr)encoding)rrdumpr )rrfrrrr!-s"zPP3DConfig.dumpcCsVd|vrtd|jddkr||jd<dS|jddkr$||jd<dS||jd<dS) N lr_schedulerzKNot able to update learning rate, because no LR scheduler config was found.typeZOneCycleZlr_maxZOneCycleWarmupDecayLrZbase_learning_rate learning_rate) RuntimeErrorr#rr%rrrupdate_learning_rate1szPP3DConfig.update_learning_ratetraincC,|dkr |d|dStdt|d)Nr) batch_sizezSetting `batch_size` in  mode is not supported.set_val ValueErrorrepr)rr+moderrrupdate_batch_size?s zPP3DConfig.update_batch_sizecCr*)Nr)epochszSetting `epochs` in r,r-)rr3r1rrr update_epochsGszPP3DConfig.update_epochsFcCstr)NotImplementedError)rZ weight_pathZ is_backbonerrrupdate_pretrained_weightsMsz$PP3DConfig.update_pretrained_weightscCs d|vr|jSd|vs J|jSNitersr3)r8r3rrrrget_epochs_itersPs zPP3DConfig.get_epochs_iterscCs@d|vs d|jvr dS|jd}t|tr|d}t|ts|S)Nr#r%g-C6?)r#r r )rlrrrrget_learning_rateWs   zPP3DConfig.get_learning_ratecCsd|vr|jSdS)Nr+)r+)rr1rrrget_batch_sizeaszPP3DConfig.get_batch_sizecCs<d|vsJdd|jvr|jdSd|jvsJ|jdS)Nfinetune_config4QAT training yaml should contain finetune_config keyr8r3r?r9rrrget_qat_epochs_itershs    zPP3DConfig.get_qat_epochs_iterscCsVd|vsJd|j}d|vsd|jvrdS|jd}t|tr)|d}t|ts |S)Nr?r@r#r%gMb ?)r?r#r r )rcfgr;rrrget_qat_learning_raters    z PP3DConfig.get_qat_learning_ratecC||jd<dS)NZ warmup_stepsr#)rZstepsrrrupdate_warmup_stepszPP3DConfig.update_warmup_stepscCrE)NZend_lrrFr'rrr update_end_lrrHzPP3DConfig.update_end_lrcCs(|d|d|vr|dddSdSr7)r.rr8rrr update_iterss zPP3DConfig.update_iterscCs&||jd<d|jvrd|jd<dSdSr7rArJrrrupdate_finetune_iterss  z PP3DConfig.update_finetune_iterssave_dircCst||d<dS)NrMr)rrMrrrupdate_save_dirszPP3DConfig.update_save_dirN)r))F)__name__ __module__ __qualname__rrr!r(r2r4r6r:r<r>rBrDrGrIrKrLstrrNrrrrrs$     r)rrZ utils.miscrbaserrrrrrs