B 1rDa*ã@s:ddlZddlZddlmZddlZdd„Zddd„ZdS) éN)Úloggerc Cs†| ¡}i}xf| ¡D]Z\}}||kr8t d |¡¡q||}|j|jkrht d ||j||j¡¡q|||<qW|j|dd|S)NzF{} is not in the ckpt. Please double check and see if this is desired.zBShape of {} in checkpoint is {}, while shape of {} in model is {}.F)Ústrict)Ú state_dictÚitemsrÚwarningÚformatÚshapeÚload_state_dict)ÚmodelZckptZmodel_state_dictÚ load_dictZ key_modelÚvZv_ckpt©r õF/home/huangxh/æ¡Œé¢/icvrcautonomous-driving/yolox/utils/checkpoint.pyÚ load_ckpt s"  rÚcCsVtj |¡st |¡tj ||d¡}t ||¡|rRtj |d¡}t ||¡dS)Nz _ckpt.pthz best_ckpt.pth) ÚosÚpathÚexistsÚmakedirsÚjoinÚtorchÚsaveÚshutilÚcopyfile)ÚstateZis_bestZsave_dirÚ model_nameÚfilenameZ best_filenamer r rÚsave_checkpoint$s   r)r)rrÚlogururrrrr r r rÚs