Ë
    ˆ\;jÎ¼  ã                   ón  — d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl	Z
d dlZd dlmZ d dlmZ d dlmZmZmZmZmZmZmZ ddlmZmZmZmZmZmZmZ g Zg Z d„ Z!d$d	„Z"d
„ Z#d„ Z$d„ Z%d„ Z&d„ Z'd„ Z(d„ Z)d„ Z*d„ Z+d„ Z,d„ Z-d„ Z.d„ Z/d„ Z0d„ Z1d„ Z2d„ Z3d„ Z4d„ Z5d„ Z6d„ Z7d„ Z8d%d „Z9d&d!„Z:d"„ Z;d#„ Z<y)'é    N)ÚIterable)Úbase)Úcore)ÚEagerParamBaseÚProgramÚVariableÚ_create_tensorÚ_current_expected_placeÚ_dygraph_tracerÚin_dygraph_modeé   )Ú_is_file_pathÚ_is_memory_bufferÚ_legacy_static_saveÚ_open_file_bufferÚ_pack_loaded_dictÚ_pickle_loads_macÚ_unpack_saved_dictc                  ó¼   — t        t        «      dkD  rJt        j                  «       } | r | j                  «       r| j	                  «        t        t        «      dkD  rŒIyy)z4
    wait until all async save task to be done.
    r   N)ÚlenÚasync_save_queueÚpopÚis_aliveÚjoin)Útasks    ú\G:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\paddle/framework/io.pyÚclear_async_save_task_queuer   7   sD   € ô ÔÓ
 !Ò
#Ü×#Ñ#Ó%ˆÙ�D—M‘M”OØ�I‰IŒKô ÔÓ
 !Õ
#ó    c                 ó  ‡— t        «       st        d«      ‚t        |«      dkD  rt        j                  d«       ˆfd„Št        | t        «      r	 ‰| «       nbt        | t        j                  j                  «      r't        j                  «       r| j                  «       n| } nt        dt        | «      › �«      ‚|r
t        «        t        j                   t"        | ||f¬«      }|j%                  «        t&        j)                  |«       y)a,  
    async version of paddle.save.
    Note:
        currently only support dygraph mode.
    Note:
        any argument passed through configs will be overrided by default setting.
    Args:
        obj(Object) : The object to be saved.
        path(str|BytesIO) : The path/buffer of the object to be saved.
          If saved in the current directory, the input path string will be used as the file name.
        protocol(int, optional): The protocol version of pickle module must be greater than 1 and less than 5.
                                 Default: 4
        sync_other_task(bool) : Determine whether to wait other async save task to be finished before this one be put in queue.
        **configs(dict, optional): compatible argument to paddle.save, but will be overrided by default setting.
    Examples:
        .. code-block:: python
            :name: code-example-1

            import paddle
            emb = paddle.nn.Embedding(10, 10)
            layer_state_dict = emb.state_dict()

            # call paddle.async_save with the same style of paddle.save
            paddle.async_save(layer_state_dict, "emb.pdparams")
            for i in range(10):
                # do some calculations here
            # wait if any async_save task has not been done
            paddle.clear_async_task_queue()
    z5async_save currently is not supported in static mode.r   zNconfigs are not supported in async mode, will be overided by default settings.c                 ó  •— | j                  «       D ]l  \  }}t        |t        «      r	 ‰|«       Œt        |t        j                  j
                  «      sŒDt        j                  «       r|j                  «       n|| |<   Œn y ©N)ÚitemsÚ
isinstanceÚdictr   ÚeagerÚTensorÚis_compiled_with_cudaÚ
pin_memory)ÚsdÚkÚvÚmove_state_dict_to_cpus      €r   r,   z*async_save.<locals>.move_state_dict_to_cpui   sY   ø€ Ø—H‘H–J‰DˆAˆqÜ˜!œTÔ"Ù& qÕ)Ü˜AœtŸz™z×0Ñ0Õ1Ü*.×*DÑ*DÔ*F˜Ÿ™œÈA��1’ñ	 r   z1currently async_save does not support this type: )ÚtargetÚargsN)r   Ú
ValueErrorr   ÚwarningsÚwarnr#   r$   r   r%   r&   r'   r(   Ú	TypeErrorÚtyper   Ú	threadingÚThreadÚsaveÚstartr   Úappend)ÚobjÚpathÚprotocolÚsync_other_taskÚconfigsÚtr,   s         @r   Ú
async_saver?   A   s×   ø€ ô< ÔÜØCó
ð 	
ô ˆ7ƒ|�aÒÜ�‰Ø\ô	
ô
Nô �#”tÔÙ˜sÕ#Ü	�CœŸ™×*Ñ*Ô	+Ü"&×"<Ñ"<Ô">ˆc�n‰nÔÀC‰ô Ø?ÄÀSÃ	¸{ÐKó
ð 	
ñ Ü#Ô%Ü×Ñ¤¨C°°xÐ+@ÔA€AØ‡G�G„IÜ×Ñ˜AÕr   c                 ó&  — i }i }| j                  «       D ]ô  \  }}t        |t        t        j                  j
                  f«      rÀ|j                  t        j                  j                  j                  k(  r"|j                  «       j                  «       ||<   n]|j                  «       j                  «       j                  «       st        d«      ‚t        j                   |j#                  «       «      ||<   |j$                  ||<   Œð|||<   Œö ||d<   |S )NzdThe saved tensor is not initialized. If you used group sharded, please use save_group_sharded_model.úStructuredToParameterName@@)r"   r#   r   r   r%   r&   r3   ÚVarDescÚVarTypeÚVOCABÚvalueÚget_map_tensorÚ
get_tensorÚ_is_initializedr/   ÚnpÚarrayÚcpuÚname)Ú
state_dictÚ	save_dictÚ
name_tableÚkeyrE   s        r   Ú_build_saved_state_dictrQ   €   sÞ   € Ø€IØ€JØ ×&Ñ&Ö(‰
ˆˆUÜ�eœh¬¯
©
×(9Ñ(9Ð:Ô;Ø�z‰zœTŸ\™\×1Ñ1×7Ñ7Ò7Ø!&§¡£×!=Ñ!=Ó!?�	˜#’à—{‘{“}×/Ñ/Ó1×AÑAÔCÜ$Ø~óð ô "$§¡¨%¯)©)«+Ó!6�	˜#‘Ø#Ÿj™jˆJ�sŠOà"ˆI�cŠNð )ð 0:€IÐ+Ñ,àÐr   c                 ó¶  — ddl m}m}  || |j                  «      }t        j
                  j                  «       5   || ||j                  d¬«      }i }|D ]+  }t        j                  ||   j                  «       «      ||<   Œ- t        |j                  «      dz   }t        j                  j                  | |«      }	t        j                  j                  |	«      r^t!        |	d«      5 }
t#        j$                  |
«      }d d d «       i }|D ]+  }|   j'                  dd «      }|€
J d|z  «       ‚||   ||<   Œ- |}d d d «       |S # 1 sw Y   ŒGxY w# 1 sw Y   S xY w)	Nr   )Ú_construct_params_and_buffersÚ_construct_program_holdersF)Úappend_suffixz.infoÚrbÚstructured_namezACannot find saved variable (%s)'s structured name in saved model.)Úpaddle.jit.translated_layerrS   rT   Úmodel_filenamer   ÚdygraphÚguardÚparams_filenamerI   rJ   rK   ÚstrÚosr:   r   ÚexistsÚopenÚpickleÚloadÚget)Ú
model_pathÚconfigrS   rT   ÚprogramsÚpersistable_var_dictÚload_param_dictÚvar_nameÚvar_info_filenameÚvar_info_pathÚfÚextra_var_infoÚstructured_para_dictrW   s                 r   Ú*_load_state_dict_from_save_inference_modelro   •   sd  € ÷ñ
 *¨*°f×6KÑ6KÓL€Hô 
�‰×	Ñ	Õ	Ù<Ø˜ &×"8Ñ"8Èô 
Ðð
 ˆÛ,ˆHÜ(*¯©Ø$ XÑ.×2Ñ2Ó4ó)ˆO˜HÒ%ð -ô   × 6Ñ 6Ó7¸'ÑAÐÜŸ™Ÿ™ ZÐ1BÓCˆÜ�7‰7�>‰>˜-Ô(Ü�m TÔ*¨aÜ!'§¡¨Q£�÷ +à#%Ð Û+�Ø"0°Ñ":×">Ñ">Ø% tó#�ð 'Ð2ð ØWØñóÐ2ð 9HØñ9Ð$ _Ò5ð ,ð 3ˆO÷= 
ð@ Ð÷! +Ð*ú÷! 
ð@ Ðús$   ºB,EÃ&EÃ<<EÅE	ÅEÅEc                 óÀ  — g }g }t        j                  | «      D ]p  \  }}}|D ]e  }t         j                  j                  ||«      }t         j                  j	                  || «      }|j                  dd«      }	|j                  |	«       Œg Œr t        j                  j                  «       5  |D ]_  }
t        |
d¬«      }t        «       j                  di d|idt         j                  j                  | |
«      i¬«       |j                  |«       Œa 	 d d d «       i }|D ]2  }t        j                  |j                  «       «      ||j                   <   Œ4 |S # 1 sw Y   ŒDxY w)	NÚ\Ú/T)rL   Úpersistablerb   ÚOutÚ	file_path)r3   ÚinputsÚoutputsÚattrs)r^   Úwalkr:   r   ÚrelpathÚreplacer8   r   rZ   r[   r	   r   Útrace_oprI   rJ   rK   rL   )rd   Úload_var_listÚvar_name_listÚrootÚ_ÚfilesÚfilenameru   Útmp_var_nameri   rL   Únew_varrh   Úvars                 r   Ú!_load_state_dict_from_save_paramsr†   Ä   s2  € ð €Mð €MÜŸ'™' *Ö-‰ˆˆa�ÛˆHÜŸ™Ÿ™ T¨8Ó4ˆIÜŸ7™7Ÿ?™?¨9°jÓAˆLØ#×+Ñ+¨D°#Ó6ˆHØ× Ñ  Õ*ñ	 ð .ô 
�‰×	Ñ	Õ	Û!ˆDÜ$¨$¸DÔAˆGÜÓ×&Ñ&ØØØ Ð(Ø"¤B§G¡G§L¡L°¸TÓ$BÐCð	 'ô ð × Ñ  Õ)ñ "÷ 
ð €OÛˆÜ$&§H¡H¨S¯W©W«YÓ$7ˆ˜Ÿ™Ò!ð ð Ð÷! 
Ð	ús   Â+A%EÅEc                 óÆ  — ddl m}m} | |z   }t        j                  j                  |«      }t        j                  j                  | «      }|r|rt        d| › d| › d�«      ‚|s_|s]d}| dz   }| dz   }	t        j                  j                  |«      st        j                  j                  |	«      r|d	z  }t        || z  «      ‚|r˜t        j                  j                  | «      }
t        j                  j                  | «      }|j                  �t        j                  d
«       |
|z   |_
        |j                  �t        j                  d«       |
|z   |_        ||fS | }||fS )Nr   )ÚINFER_MODEL_SUFFIXÚINFER_PARAMS_SUFFIXzThe z.pdmodel and z‚ directory exist at the same time, don't know which one to load, please make sure that the specified target of ``path`` is unique.z+The ``path`` (%s) to load model not exists.z	.pdparamsz.pdoptz©please specify the full file name, not just the file name prefix. For example, it should be written as `paddle.load('model.pdparams')` instead of `paddle.load('model')`.zqWhen loading the result saved with the specified file prefix, the ``model_filename`` config does not take effect.zrWhen loading the result saved with the specified file prefix, the ``params_filename`` config does not take effect.)rX   rˆ   r‰   r^   r:   r_   Úisdirr/   ÚbasenameÚdirnamerY   r0   r1   r\   )r:   re   rˆ   r‰   Úprefix_format_pathÚprefix_format_existÚdirectory_format_existÚ	error_msgÚparams_file_pathÚopti_file_pathÚfile_prefixrd   s               r   Ú_build_load_path_and_configr”   ô   sl  € ÷
ð
 Ð 2Ñ2ÐÜŸ'™'Ÿ.™.Ð);Ó<ÐÜŸW™WŸ]™]¨4Ó0ÐÙÑ5ÜØ�4�&˜ d Vð ,%ð %ó
ð 	
ñ
 !Ñ)?ØAˆ	ð   +Ñ-ÐØ ™ˆÜ�7‰7�>‰>Ð*Ô+¬r¯w©w¯~©~¸nÔ/MØð*ñˆIô
 ˜ TÑ)Ó*Ð*áÜŸ'™'×*Ñ*¨4Ó0ˆKÜŸ™Ÿ™¨Ó.ˆJØ×$Ñ$Ð0Ü—‘ð'ôð
 %0Ð2DÑ$DˆFÔ!Ø×%Ñ%Ð1Ü—‘ð'ôð
 &1Ð3FÑ%FˆFÔ"ð
 �vÐÐð ˆJà�vÐÐr   c                 ó  — g d¢}| D ]  }||vsŒt        d|z  «      ‚ ddlm}  |«       }| j                  dd «      |_        | j                  dd «      |_        | j                  dd «      |_        | j                  dd	«      |_        |S )
N)rY   r\   Úkeep_name_tableÚreturn_numpyz=The additional config (%s) of `paddle.load` is not supported.r   ©Ú_SaveLoadConfigrY   r\   r–   r—   F)r/   Úpaddle.jit.apir™   rc   rY   r\   r–   r—   ©r=   Úsupported_configsrP   r™   Úinner_configs        r   Ú_parse_load_configrž   .  s™   € òÐó ˆØÐ'Ò'ÜØOØñóð ð õ /á"Ó$€LØ")§+¡+Ð.>ÀÓ"E€LÔØ#*§;¡;Ð/@À$Ó#G€LÔ Ø#*§;¡;Ð/@À$Ó#G€LÔ Ø '§¡¨N¸EÓ B€LÔàÐr   c                 ó¶   — ddg}| D ]  }||vsŒt        d|z  «      ‚ ddlm}  |«       }| j                  dd«      |_        | j                  dd «      |_        |S )NÚuse_binary_formatÚpickle_protocolz=The additional config (%s) of `paddle.save` is not supported.r   r˜   F)r/   rš   r™   rc   r    r¡   r›   s        r   Ú_parse_save_configr¢   L  sv   € Ø,Ð.?Ð@Ðó ˆØÐ'Ò'ÜØOØñóð ð õ /á"Ó$€LØ%,§[¡[Ð1DÀeÓ%L€LÔ"Ø#*§;¡;Ð/@À$Ó#G€LÔ àÐr   c                 óh  ‡
‡‡‡— t        |t        «      st        dt        |«      › �«      ‚|dk  s|dkD  rt        d|› �«      ‚d„ Šd„ Šd„ Ši Š
ˆ
ˆfd„}t	        | d	„ |«       ˆ
ˆˆfd
„}ˆ
fd„}t
        j                  dk(  rut
        j                  j                  dk(  rX |«        t        j                  | «      } |«        d}t        dt        |«      |«      D ]  }|j                  ||||z    «       Œ y t        j                  ||«      }	t        j                   j#                  «       |	_        ‰|	j                   t$        j&                  <   ‰|	j                   t$        j(                  j*                  <   ‰|	j                   t,        <   |	j                   j/                  ‰
«       |	j1                  | «       y )Nú+The 'protocol' MUST be `int`, but received é   é   ú/Expected 1<'protocol'<5, but received protocol=c                 óv   — t        j                  | j                  «       «      }| j                  }t        ||fffS r!   )rI   rJ   rK   rL   Útuple)ÚselfÚdatarL   s      r   Úreduce_varbasez$_pickle_save.<locals>.reduce_varbaseo  s0   € Ü�x‰x˜Ÿ™›
Ó#ˆØ�y‰yˆä˜˜t˜�Ð'Ð'r   c                 óÐ   — t        j                  «       }|j                  t        j                  «       «       t        j                  | j                  |«      «      }t        dd|iffS )Nr«   )	r   ÚPlaceÚ	set_placeÚpaddleÚCPUPlacerI   rJ   Ú_copyÚeval)rª   Úpr«   s      r   Úreduce_LoDTensorz&_pickle_save.<locals>.reduce_LoDTensoru  sJ   € Ü�J‰J‹LˆØ	�‰”F—O‘OÓ%Ô&Ü�x‰x˜Ÿ
™
 1›Ó&ˆä�v ¨˜~Ð.Ð/Ð/r   c                 ó   — t        d«      ‚)Nz6paddle do not support saving `paddle.nn.Layer` object.)r/   )rª   s    r   Úreduce_Layerz"_pickle_save.<locals>.reduce_Layer|  s   € ÜØDó
ð 	
r   c                 ó&   •— ‰‰| j                   <   | S r!   )Ú	__class__)ÚlayerÚdispatch_table_layerr·   s    €€r   Úcreate_layer_dispatch_tablez1_pickle_save.<locals>.create_layer_dispatch_tableƒ  s   ø€ Ø0<Ð˜UŸ_™_Ñ-Øˆr   c                 óJ   — t        | t        j                  j                  «      S r!   ©r#   r°   ÚnnÚLayer)r+   s    r   Ú<lambda>z_pickle_save.<locals>.<lambda>‰  s   € ”*˜Q¤§	¡	§¡Ô0r   c                  ó
  •— ‰t         j                  t        j                  j                  <   ‰t         j                  t
        <   ‰t         j                  t        j                  <   t         j                  j                  ‰ «       y r!   )ra   Údispatch_tabler   r%   r&   r   Ú	LoDTensorÚupdate)r»   rµ   r¬   s   €€€r   Úadd_dispatch_tablez(_pickle_save.<locals>.add_dispatch_table�  sV   ø€ à3AŒ×ÑœdŸj™j×/Ñ/Ñ0Ø0>Œ×ÑœnÑ-Ø0@Œ×ÑœdŸn™nÑ-Ü×Ñ×$Ñ$Ð%9Õ:r   c                  ó`  •— t         j                  j                  t        j                  «       t         j                  j                  t        j
                  j                  «       t         j                  j                  t        «       ‰D ]!  } t         j                  j                  | «       Œ# y r!   )ra   rÃ   r   r   rÄ   r%   r&   r   )r*   r»   s    €r   Úpop_dispatch_tablez(_pickle_save.<locals>.pop_dispatch_table”  sk   ø€ Ü×Ñ×!Ñ!¤$§.¡.Ô1Ü×Ñ×!Ñ!¤$§*¡*×"3Ñ"3Ô4Ü×Ñ×!Ñ!¤.Ô1Û%ˆAÜ×!Ñ!×%Ñ% aÕ(ñ &r   Údarwiné   é   @r   )r#   Úintr/   r3   Ú_parse_every_objectÚsysÚplatformÚversion_infoÚmajorra   ÚdumpsÚranger   ÚwriteÚPicklerÚcopyregrÃ   Úcopyr   rÄ   r%   r&   r   rÅ   Údump)r9   rl   r;   r¼   rÆ   rÈ   Úpickle_bytesÚ	max_bytesÚiÚpicklerr»   r·   rµ   r¬   s             @@@@r   Ú_pickle_saverÝ   c  s  û€ ä�h¤Ô$ÜØ9¼$¸x».Ð9IÐJó
ð 	
ð �!‚|�x !’|ÜØ=¸h¸ZÐHó
ð 	
ò(ò0ò
ð
 Ðõô ØÙ0Ø#ôö;ô)ô ‡|�|�xÒ¤C×$4Ñ$4×$:Ñ$:¸aÒ$?ÙÔÜ—|‘| CÓ(ˆÙÔàˆ	Ü�qœ#˜lÓ+¨YÖ7ˆAØ�G‰G�L  Q¨¡]Ð3Õ4ñ 8ô —.‘.  HÓ-ˆÜ!(×!7Ñ!7×!<Ñ!<Ó!>ˆÔà1Aˆ×ÑœtŸ~™~Ñ.Ø4Bˆ×ÑœtŸz™z×0Ñ0Ñ1Ø1?ˆ×Ñœ~Ñ.Ø×Ñ×%Ñ%Ð&:Ô;Ø�‰�SÕr   c                 ó–  — t        | t        j                  «      rt        d«      ‚ || «      ryt	        | «      t
        t        j                  t        t        fv rpt	        | «      t
        t        j                  fv rt        | j                  «       «      }nt        t        | «      «      }d}|D ]  }|t        | |   |«      z  }|sŒ y |S y)Nz3`paddle.save` do not support saving 'SelectedRows'.TF)r#   r   ÚSelectedRowsÚNotImplementedErrorr3   r$   ÚcollectionsÚOrderedDictÚlistr©   ÚkeysrÓ   r   Ú
_contain_x)r9   Úcondition_funcrä   ÚflagrP   s        r   rå   rå   ¯  s²   € Ü�#”t×(Ñ(Ô)Ü!ØAó
ð 	
ñ �cÔØÜ	ˆc‹”tœ[×4Ñ4´d¼EÐBÑ	BÜ�‹9œœ{×6Ñ6Ð7Ñ7Ü˜Ÿ™›
Ó#‰Däœ˜S›“?ˆDØˆÛˆCØ”J˜s 3™x¨Ó8Ñ8ˆDÚÙð ð ˆàr   c                 ó8  — t        | t        «      rŠd„ }| j                  «       D ]s  \  }}t        |t        «      r(|j                  «       D ]  \  }}t        ||«      sŒ  y Œ>t        |t        j
                  j                  t        j                  f«      rŒs y yy)Nc                 óÄ   — t        | t        j                  j                  t        t
        j                  j                  t
        j                  t
        j                  f«      S r!   )
r#   r°   r¿   rÀ   r   r   r%   r&   rÄ   rß   ©r9   s    r   Ú	conditionz!_is_state_dict.<locals>.conditionÉ  sB   € ÜØä—I‘I—O‘OÜÜ—J‘J×%Ñ%Ü—N‘NÜ×%Ñ%ðó	ð 	r   FT)r#   r$   r"   rå   r   r%   r&   rÄ   )r9   rë   rP   rE   r*   r+   s         r   Ú_is_state_dictrì   Æ  s|   € Ü�#”tÔò
	ð Ÿ)™)ž+‰JˆC�Ü˜%¤Ô&Ø!ŸK™KžM‘D�A�qÜ! ! YÕ/Ú$ñ *ô   ¬¯
©
×(9Ñ(9¼4¿>¹>Ð'JÕKÙð &ð àr   c                 ó¦   — t        | t        «      rAt        | «      dk(  r3t        }t        | d   |«      rt        | d   t        j
                  «      ryy)Nr¥   r   r   TF)r#   r©   r   r]   rI   Úndarray)r9   Ú
name_typess     r   Ú_transformed_from_varbaserð   ä  sB   € ô �#”uÔ¤# c£(¨a¢-Üˆ
Ü�c˜!‘f˜jÔ)¬j¸¸Q¹ÄÇÁÔ.LØØr   c                 ó:   — t        | t        j                  «      ryy)NTF)r#   rI   rî   rê   s    r   Ú_transformed_from_lodtensorrò   î  s   € ô �#”r—z‘zÔ"ØØr   c                 óÊ   — t        | t        j                  «      st        dt	        | «      › d�«      ‚t        j                  «       }t        «       }|j                  | |«       |S )Nz8Type of `ndarray` should be numpy.ndarray, but received Ú.)	r#   rI   rî   r2   r3   r   rÄ   r
   Úset)rî   r>   Úplaces      r   Ú_to_LodTensorr÷   ö  sW   € Ü�gœrŸz™zÔ*ÜØFÄtÈGÃ}ÀoÐUVÐWó
ð 	
ô 	�‰Ó€AÜ#Ó%€EØ‡E�Eˆ'�5ÔØ€Hr   c                 óˆ   — |r| d   S t        «       r$t        j                  | d   «      }| d   |_        |S t	        | d   «      S )Nr   r   )r   r°   Ú	to_tensorrL   r÷   )r9   r—   r>   s      r   Ú_tuple_to_tensorrú     sI   € ÙØ�1‰vˆÜÔÜ×Ñ˜S ™VÓ$ˆð �Q‘ˆŒØˆä˜S ™VÓ$Ð$r   c                 ó^   — |r| S t        «       rt        j                  | «      S t        | «      S r!   )r   r°   rù   r÷   ©r9   r—   s     r   Ú_ndarray_to_tensorrý     s,   € ÙØˆ
ÜÔÜ×Ñ Ó$Ð$ä˜SÓ!Ð!r   c                 ó†   — t        «       }|j                  «       j                  «       j                  | t	        «       «       |S r!   )r	   rE   rG   rõ   r
   )ÚtensorÚ
return_vars     r   Ú_lod_tensor2varbaser    s6   € ÜÓ!€JØ×ÑÓ×!Ñ!Ó#×'Ñ'¨Ô0GÓ0IÔJØÐr   c                 ó  —  || «      r || «      S t        | «      t        t        j                  t        fv rvt        | «      t        k(  rt        t        | «      «      }nt	        | j                  «       «      }|D ]/  } || |   «      r || |   «      | |<   Œt        | |   ||«      | |<   Œ1 | S t        | «      t        k(  rt        t        t	        | «      ||«      «      S t        | «      t        k(  rt        t        t	        | «      ||«      «      S t        | t        «      rkt        | t        t        j                  t         j"                  j$                  t         j&                  f«      s#t)        dj+                  t        | «      «      «      ‚| S )Nz]The iteratable objects supported are tuple, list, dict, OrderedDict, string. But received {}.)r3   r$   rá   râ   rã   rÓ   r   rä   rÍ   r©   rõ   r#   r   r]   rI   rî   r   r%   r&   rÄ   rà   Úformat)r9   ræ   Úconvert_funcrä   rP   s        r   rÍ   rÍ     sL  € Ù�cÔÙ˜CÓ Ð Ü	ˆc‹”tœ[×4Ñ4´dÐ;Ñ	;Ü�‹9œÒÜœ˜S›“?‰Dä˜Ÿ™›
Ó#ˆDÛˆCÙ˜c #™hÔ'Ù'¨¨C©Ó1��C’ä.Ø˜‘H˜n¨ló��C’ð	 ð ˆ
Ü	ˆc‹”eÒ	ÜÜ¤ S£	¨>¸<ÓHó
ð 	
ô 
ˆc‹”cÒ	ÜÔ&¤t¨C£y°.À,ÓOÓPÐPä�cœ8Ô$¬ZØÜ”"—*‘*œdŸj™j×/Ñ/´·±Ð@ô.
ô &Øo×vÑvÜ˜“Ióóð ð
 ˆ
r   c                 óä   ‡— d„ }d„ }t        | |«      r"t        «       st        d«      ‚t        | ||«       ˆfd„}ˆfd„}t        | t        «      rt        | t        |«      S t        | t
        |«      S )Nc                 óJ   — t        | t        j                  j                  «      S r!   r¾   rê   s    r   Úis_layerz$_parse_load_result.<locals>.is_layerA  s   € Ü˜#œvŸy™yŸ™Ó/Ð/r   c                 óh   — t        | j                  d«      }| j                  j                  |«       | S )NF)Ú_parse_load_resultÚ__dict__rÅ   )r9   Ú	temp_dicts     r   Úparse_layerz'_parse_load_result.<locals>.parse_layerD  s)   € Ü& s§|¡|°UÓ;ˆ	Ø�‰×Ñ˜IÔ&Øˆ
r   zMLayer can only be loaded in dynamic graph mode, but now in static graph mode.c                 ó   •— t        | ‰¬«      S ©N)r—   )rú   rü   s    €r   Útuple_to_tensorz+_parse_load_result.<locals>.tuple_to_tensorQ  s   ø€ Ü °,Ô?Ð?r   c                 ó   •— t        | ‰¬«      S r  )rý   rü   s    €r   Úndarray_to_tensorz-_parse_load_result.<locals>.ndarray_to_tensorT  s   ø€ Ü! #°LÔAÐAr   )rå   r   r/   rÍ   rð   rò   )r9   r—   r  r  r  r  s    `    r   r	  r	  @  sƒ   ø€ ò0òô
 �#�xÔ ÜÔ ÜØ_óð ô 	˜C ¨;Ô7ô@ôBô
 �#Ô0Ô1Ü"ØÔ*¨Oó
ð 	
ô #ØÔ,Ð.?ó
ð 	
r   c                 óŽ  — | j                  «       st        d«      ‚t        |«      rt        j                  | |«      }|S t        |«      rMt        j                  | «      }t        |d«      5 }|j                  |«       |j                  «       }d d d «       |S t        dj                  t        |«      «      «      ‚# 1 sw Y   S xY w)NzlThe saved tensor is not initialized. If you used group sharded, please use save_group_sharded_model firstly.Úwbú@Only supports saving objects to file or BytesIO, but received {})rH   r/   r   r   Úsave_lod_tensorr   Úsave_lod_tensor_to_memoryr   rÔ   Útellrà   r  r3   )rÿ   Ú	file_nameÚ_seekÚtensor_bytesrl   s        r   Ú_save_lod_tensorr  e  s»   € Ø×!Ñ!Ô#ÜØzó
ð 	
ô �YÔÜ×$Ñ$ V¨YÓ7ˆð  €Lô 
˜9Ô	%Ü×5Ñ5°fÓ=ˆä˜y¨$Ô/°1Ø�G‰G�LÔ!Ø—F‘F“HˆE÷ 0ð €Lô "ØN×UÑUÜ�Y“óó
ð 	
÷ 0ð €Lús   Á+"B:Â:Cc                 ó  — t         j                  j                  j                  «       }t	        | «      r.t         j                  j                  j                  || «      }||fS t        | «      rct        | d«      5 }|j                  «       }t         j                  j                  j                  ||«       |j                  «       }d d d «       ||fS t        dj                  t        | «      «      «      ‚# 1 sw Y   |fS xY w©NrV   z@Only supports load objects from file or BytesIO, but received {})r°   r   r   rÄ   r   Úload_lod_tensorr   r   ÚreadÚload_lod_tensor_from_memoryr  rà   r  r3   )r  Útemp_tr  rl   r  s        r   Ú_load_lod_tensorr"  ~  s×   € Ü�[‰[×Ñ×'Ñ'Ó)€FÜ�YÔä—‘× Ñ ×0Ñ0°¸ÓCˆð �5ˆ=Ðô 
˜9Ô	%Ü˜y¨$Ô/°1ØŸ6™6›8ˆLÜ�K‰K×Ñ×8Ñ8¸ÀÔNØ—F‘F“HˆE÷ 0ð �5ˆ=Ðô "ØN×UÑUÜ�Y“óó
ð 	
÷ 0ð �5ˆ=Ðús   Á9AC3Ã3C?c                 óª  — | j                  «       j                  «       st        d«      ‚t        |«      rt	        j
                  | |«      }|S t        |«      rMt	        j                  | «      }t        |d«      5 }|j                  |«       |j                  «       }d d d «       |S t        dj                  t        |«      «      «      ‚# 1 sw Y   S xY w)Nz$The saved tensor is not initialized.r  r  )rG   rH   r/   r   r   Úsave_selected_rowsr   Úsave_selected_rows_to_memoryr   rÔ   r  rà   r  r3   )Úselected_rowsr  r  Úselected_rows_bytesrl   s        r   Ú_save_selected_rowsr(  ”  sÀ   € Ø×#Ñ#Ó%×5Ñ5Ô7ÜÐ?Ó@Ð@Ü�YÔä×'Ñ'¨°yÓAˆð €Lô 
˜9Ô	%Ü"×?Ñ?ÀÓNÐÜ˜y¨$Ô/°1Ø�G‰GÐ'Ô(Ø—F‘F“HˆE÷ 0ð €Lô "ØN×UÑUÜ�Y“óó
ð 	
÷	 0ð €Lús   Á9"CÃCc                 ó®  — t        j                  «       }t        | «      rt        j                  || «      }||fS t	        | «      rct        | d«      5 }|j                  «       }t        j                  j                   j                  ||«       d d d «       j                  «       }||fS t        dj                  t        | «      «      «      ‚# 1 sw Y   Œ@xY wr  )r   rß   r   Úload_selected_rowsr   r   r  r°   r   Úload_selected_rows_from_memoryr  rà   r  r3   )r  Útemp_srr  rl   r'  s        r   Ú_load_selected_rowsr-  ©  sÁ   € Ü×ÑÓ!€GÜ�YÔä×'Ñ'¨°Ó;ˆð" �Eˆ>Ðô 
˜9Ô	%Ü˜y¨$Ô/°1Ø"#§&¡&£(ÐÜ�K‰K×Ñ×;Ñ;ØÐ,ô÷ 0ð
 —‘“ˆð �Eˆ>Ðô "ØN×UÑUÜ�Y“óó
ð 	
÷ 0Ð/ús   Á;CÃCc                 ó~  — t        | t        j                  «      rt        | |«       y t        | t        j                  «      rt        | |«       y t        | t        j                  j                  «      r)t        | j                  «       j                  «       |«       y t        dj                  t        | «      «      «      ‚)NzOWhen use_binary_format = True, `paddle.save`  expected Tensor, but received {}.)r#   r   rÄ   r  rß   r(  r%   r&   rE   rG   rà   r  r3   )r9   r:   s     r   Ú_save_binary_varr/  Á  s…   € Ü�#”t—~‘~Ô&Ü˜˜dÕ#Ü	�Cœ×*Ñ*Ô	+Ü˜C Õ&Ü	�CœŸ™×*Ñ*Ô	+Ü˜Ÿ™›×/Ñ/Ó1°4Õ8ô "Ø]×dÑdÜ�S“	óó
ð 	
r   c                 óD  — t        |«      r‡t        j                  j                  |«      }|dk(  rt	        d«      ‚t        j                  j                  |«      }|rYt        j                  j                  |«      s:t        j                  |d¬«       n"t        |«      st	        dt        |«      › �«      ‚t        |«      }t        |j                  t        «      s-t        dj                  t        |j                  «      «      «      ‚|j                  rt!        | |«       y|j"                  �!|j"                  }t%        j&                  d«       t        | t(        «      rY| j*                  j-                  «        t/        |d	«      5 }|j1                  | j*                  j3                  «       «       ddd«       yt5        | «      r&t7        «       rt9        | ||«       yt;        | ||«       yt/        |d	«      5 }t=        | ||«       ddd«       y# 1 sw Y   yxY w# 1 sw Y   yxY w)
a‚  
    Save an object to the specified path.

    Note:
        Now supports saving ``state_dict`` of Layer/Optimizer, Tensor and nested structure containing Tensor, Program.

    Note:
        Different from ``paddle.jit.save``, since the save result of ``paddle.save`` is a single file,
        there is no need to distinguish multiple saved files by adding a suffix. The argument ``path``
        of ``paddle.save`` will be directly used as the saved file name instead of a prefix.
        In order to unify the saved file name format, we recommend using the paddle standard suffix:
        1. for ``Layer.state_dict`` , recommend to use ``.pdparams`` ;
        2. for ``Optimizer.state_dict`` , recommend to use ``.pdopt`` .
        For specific examples, please refer to API code examples.

    Args:
        obj(Object) : The object to be saved.
        path(str|BytesIO) : The path/buffer of the object to be saved.
          If saved in the current directory, the input path string will be used as the file name.
        protocol(int, optional): The protocol version of pickle module must be greater than 1 and less than 5.
                                 Default: 4
        **configs(dict, optional): optional keyword arguments. The following options are currently supported:
          use_binary_format(bool): When the saved object is static graph variable, you can specify ``use_binary_for_var``.
          If True, save the file in the c++ binary format when saving a single static graph variable; otherwise, save it in pickle format.
          Default: False

    Returns:
        None

    Examples:
        .. code-block:: python
            :name: code-example-1

            >>> # example 1: dynamic graph
            >>> import paddle
            >>> emb = paddle.nn.Embedding(10, 10)
            >>> layer_state_dict = emb.state_dict()

            >>> # save state_dict of emb
            >>> paddle.save(layer_state_dict, "emb.pdparams")

            >>> scheduler = paddle.optimizer.lr.NoamDecay(
            ...     d_model=0.01, warmup_steps=100, verbose=True)
            >>> adam = paddle.optimizer.Adam(
            ...     learning_rate=scheduler,
            ...     parameters=emb.parameters())
            >>> opt_state_dict = adam.state_dict()

            >>> # save state_dict of optimizer
            >>> paddle.save(opt_state_dict, "adam.pdopt")
            >>> # save weight of emb
            >>> paddle.save(emb.weight, "emb.weight.pdtensor")

        .. code-block:: python
            :name: code-example-2

            >>> # example 2: Save multiple state_dict at the same time
            >>> import paddle
            >>> from paddle import nn
            >>> from paddle.optimizer import Adam

            >>> layer = paddle.nn.Linear(3, 4)
            >>> adam = Adam(learning_rate=0.001, parameters=layer.parameters())
            >>> obj = {'model': layer.state_dict(), 'opt': adam.state_dict(), 'epoch': 100}
            >>> path = 'example/model.pdparams'
            >>> paddle.save(obj, path)

        .. code-block:: python
            :name: code-example-3

            >>> # example 3: static graph
            >>> import paddle
            >>> import paddle.static as static

            >>> paddle.enable_static()

            >>> # create network
            >>> x = paddle.static.data(name="x", shape=[None, 224], dtype='float32')
            >>> z = paddle.static.nn.fc(x, 10)

            >>> place = paddle.CPUPlace()
            >>> exe = paddle.static.Executor(place)
            >>> exe.run(paddle.static.default_startup_program())
            >>> prog = paddle.static.default_main_program()
            >>> for var in prog.list_vars():
            ...     if list(var.shape) == [224, 10]:
            ...         tensor = var.get_value()
            ...         break

            >>> # save/load tensor
            >>> path_tensor = 'temp/tensor.pdtensor'
            >>> paddle.save(tensor, path_tensor)

            >>> # save/load state_dict
            >>> path_state_dict = 'temp/model.pdparams'
            >>> paddle.save(prog.state_dict("param"), path_tensor)

        .. code-block:: python
            :name: code-example-4

            >>> # example 4: save program
            >>> import paddle

            >>> paddle.enable_static()

            >>> data = paddle.static.data(
            ...     name='x_static_save', shape=(None, 224), dtype='float32')
            >>> y_static = z = paddle.static.nn.fc(data, 10)
            >>> main_program = paddle.static.default_main_program()
            >>> path = "example/main_program.pdmodel"
            >>> paddle.save(main_program, path)

        .. code-block:: python
            :name: code-example-5

            >>> # example 5: save object to memory
            >>> from io import BytesIO
            >>> import paddle
            >>> from paddle.nn import Linear
            >>> paddle.disable_static()

            >>> linear = Linear(5, 10)
            >>> state_dict = linear.state_dict()
            >>> byio = BytesIO()
            >>> paddle.save(state_dict, byio)
            >>> paddle.seed(2023)
            >>> tensor = paddle.randn([2, 3], dtype='float32')
            >>> paddle.save(tensor, byio)

    Ú ú~The input path MUST be format of dirname/filename [dirname\filename in Windows system], but received filename is empty string.T©Úexist_okz<only supports saving objects to file and `BytesIO`, but got z<Type of `use_binary_format` should be bool, but received {}.NzJ'pickle_protocol' is a deprecated argument. Please use 'protocol' instead.r  )r   r^   r:   r‹   r/   rŒ   r_   Úmakedirsr   r3   r¢   r#   r    Úboolr2   r  r/  r¡   r0   r1   r   ÚdescÚflushr   rÔ   Úserialize_to_stringrì   r   Ú_legacy_saver   rÝ   )r9   r:   r;   r=   r‚   rŒ   re   rl   s           r   r6   r6   Ñ  sµ  € ôF �TÔä—7‘7×#Ñ# DÓ)ˆØ�rŠ>Üð,óð ô —'‘'—/‘/ $Ó'ˆÙœ2Ÿ7™7Ÿ>™>¨'Ô2Ü�K‰K˜¨$Ö/Ü˜tÔ$ÜØJÌ4ÐPTË:È,ÐWó
ð 	
ô   Ó(€Fä�f×.Ñ.´Ô5ÜØJ×QÑQÜ�V×-Ñ-Ó.óó
ð 	
ð ×ÒÜ˜˜dÕ#ð ×!Ñ!Ð-Ø×-Ñ-ˆHÜ�M‰MØ\ôô �cœ7Ô#Ø�H‰H�N‰NÔÜ" 4¨Ô.°!Ø—‘˜Ÿ™×4Ñ4Ó6Ô7÷ /Ð.ô ˜CÔ ÜÔ Ü˜S $¨Õ1ä# C¨¨xÕ8ä" 4¨Ô.°!Ü˜S ! XÔ.÷ /Ð.÷ /Ð.ú÷ /Ð.ús   Æ*H
Ç3HÈ
HÈHc                 óJ  — t        | t        «      st        dt        | «      z  «      ‚t	        | «      dk(  rt        j                  d«       t        |t        «      st        dt        |«      › �«      ‚|dk  s|dkD  rt        d|› �«      ‚t        |«      r†t        j                  j                  |«      }|dk(  rt        d	«      ‚t        j                  j                  |«      }|r6t        j                  j                  |«      st        j                  |d
¬«       t        | t        «      rt!        | «      }t#        |«      }t        |«      r’t$        j&                  dk(  rt$        j(                  j*                  dk(  rbt-        j.                  ||¬«      }t1        |d«      5 }d}t3        dt	        |«      |«      D ]  }	|j5                  ||	|	|z    «       Œ 	 d d d «       y t7        |d«      5 }t-        j8                  |||¬«       d d d «       y # 1 sw Y   y xY w# 1 sw Y   y xY w)NzVNow only supports save state_dict of Layer or Optimizer, expect dict, but received %s.r   z/The input state dict is empty, no need to save.r¤   r¥   r¦   r§   r1  r2  Tr3  rÉ   rÊ   )r;   r  rË   )r#   r$   rà   r3   r   r0   r1   rÌ   r/   r   r^   r:   r‹   rŒ   r_   r5  rQ   r   rÎ   rÏ   rÐ   rÑ   ra   rÒ   r`   rÓ   rÔ   r   rØ   )
r9   r:   r;   r‚   rŒ   Ú	saved_objrÙ   rl   rÚ   rÛ   s
             r   r:  r:  ‰  sÕ  € ä�cœ4Ô Ü!ð,Ü.2°3«iñ8ó
ð 	
ô
 ˆ3ƒx�1‚}Ü�‰ÐGÔHä�h¤Ô$ÜØ9¼$¸x».Ð9IÐJó
ð 	
ð �!‚|�x !’|ÜØ=¸h¸ZÐHó
ð 	
ô �TÔÜ—7‘7×#Ñ# DÓ)ˆØ�rŠ>Üð,óð ô —'‘'—/‘/ $Ó'ˆÙœ2Ÿ7™7Ÿ>™>¨'Ô2Ü�K‰K˜¨$Õ/ä�#”tÔÜ+¨CÓ0ˆ	ä" 9¨hÓ7€Iô 	�dÔÜ�L‰L˜HÒ$Ü×Ñ×"Ñ" aÒ'ä—|‘| I¸ÔAˆÜ�$˜Ô ØˆIÜ˜1œc ,Ó/°Ö;�Ø—‘˜ Q¨¨Y©Ð7Õ8ñ <÷ Ðô
 ˜t TÔ*¨aÜ�K‰K˜	 1¨xÕ8÷ +Ð*÷ Ðú÷
 +Ð*ús   Æ 5HÇ+HÈHÈH"c           	      óî  — t        | «      s t        j                  j                  | «      �rot	        |«      }t
        j                  }	 t        | d«      5 }t        | «      r=t        j                  dk(  r*t        j                  j                  dk(  rt        | |«      }nt        j                  |d¬«      }t        |t         «      r·t#        |«      }d|v r‘|d   j%                  «       D ]g  \  }}t        ||   t&        j(                  «      sŒ$t+        ||   |j,                  «      ||<   |j,                  rŒMt/        ||   dd«      sŒ^|||   _        Œi |j2                  s5d|v r1|d= n-t5        ||j,                  «      }nt5        ||j,                  «      }d	d	d	«       S tU        | fi |¤Ž}|S # 1 sw Y   ŒxY w# |$ �r9}	 t7        | «      \  }	}
|	cY d	}~S #  	 t9        | «      \  }	}
|j,                  rbt;        j<                  «       }|j?                  tA        jB                  «       «       t'        jD                  |	jG                  |«      «      cY cY d	}~S tI        «       rtK        |	«      cY cY d	}~S |	cY cY d	}~S #  	 t        | d«      5 }|jM                  «       }tO        jP                  |«      }|cd	d	d	«       cY cY cY d	}~S # 1 sw Y   nxY wn#  tS        d
| › d�«      ‚xY wY nxY wY nxY wY d	}~S d	}~ww xY w)a§  
    Load an object can be used in paddle from specified path.

    Note:
        Now supports loading ``state_dict`` of Layer/Optimizer, Tensor and nested structure containing Tensor, Program.

    Note:
        In order to use the model parameters saved by paddle more efficiently,
        ``paddle.load`` supports loading ``state_dict`` of Layer from the result of
        other save APIs except ``paddle.save`` , but the argument ``path`` format is
        different:
        1. loading from ``paddle.static.save`` or ``paddle.Model().save(training=True)`` ,
        ``path`` needs to be a complete file name, such as ``model.pdparams`` or
        ``model.pdopt`` ;
        2. loading from ``paddle.jit.save`` or ``paddle.static.save_inference_model``
        or ``paddle.Model().save(training=False)`` , ``path`` need to be a file prefix,
        such as ``model/mnist``, and ``paddle.load`` will get information from
        ``mnist.pdmodel`` and ``mnist.pdiparams`` ;
        3. loading from paddle 1.x APIs ``paddle.base.io.save_inference_model`` or
        ``paddle.base.io.save_params/save_persistables`` , ``path`` need to be a
        directory, such as ``model`` and model is a directory.

    Note:
        If you load ``state_dict`` from the saved result of static graph mode API such as
        ``paddle.static.save`` or ``paddle.static.save_inference_model`` ,
        the structured variable name in dynamic mode will cannot be restored.
        You need to set the argument ``use_structured_name=False`` when using
        ``Layer.set_state_dict`` later.

    Args:
        path(str|BytesIO) : The path/buffer to load the target object. Generally, the path is the target
            file path. When loading state_dict from the saved result of the API used to save
            the inference model, the path may be a file prefix or directory.
        **configs (dict, optional): other load configuration options for compatibility. We do not
            recommend using these configurations, they may be removed in the future. If not necessary,
            DO NOT use them. Default None.
            The following options are currently supported:
            (1) model_filename (str): The inference model file name of the paddle 1.x
            ``save_inference_model`` save format. Default file name is :code:`__model__` .
            (2) params_filename (str): The persistable variables file name of the paddle 1.x
            ``save_inference_model`` save format. No default file name, save variables separately
            by default.
            (3) return_numpy(bool): If specified as True, return tensor as numpy.ndarray, otherwise return tensor as paddle.Tensor.
            Default False.

    Returns:
        Object(Object): a target object can be used in paddle

    Examples:
        .. code-block:: python
            :name: code-example-1

            >>> # example 1: dynamic graph
            >>> import paddle
            >>> emb = paddle.nn.Embedding(10, 10)
            >>> layer_state_dict = emb.state_dict()

            >>> # save state_dict of emb
            >>> paddle.save(layer_state_dict, "emb.pdparams")

            >>> scheduler = paddle.optimizer.lr.NoamDecay(
            ...     d_model=0.01, warmup_steps=100, verbose=True)
            >>> adam = paddle.optimizer.Adam(
            ...     learning_rate=scheduler,
            ...     parameters=emb.parameters())
            >>> opt_state_dict = adam.state_dict()

            >>> # save state_dict of optimizer
            >>> paddle.save(opt_state_dict, "adam.pdopt")
            >>> # save weight of emb
            >>> paddle.save(emb.weight, "emb.weight.pdtensor")

            >>> # load state_dict of emb
            >>> load_layer_state_dict = paddle.load("emb.pdparams")
            >>> # load state_dict of optimizer
            >>> load_opt_state_dict = paddle.load("adam.pdopt")
            >>> # load weight of emb
            >>> load_weight = paddle.load("emb.weight.pdtensor")

        .. code-block:: python
            :name: code-example-2

            >>> # example 2: Load multiple state_dict at the same time
            >>> import paddle
            >>> from paddle import nn
            >>> from paddle.optimizer import Adam

            >>> layer = paddle.nn.Linear(3, 4)
            >>> adam = Adam(learning_rate=0.001, parameters=layer.parameters())
            >>> obj = {'model': layer.state_dict(), 'opt': adam.state_dict(), 'epoch': 100}
            >>> path = 'example/model.pdparams'
            >>> paddle.save(obj, path)
            >>> obj_load = paddle.load(path)

        .. code-block:: python
            :name: code-example-3

            >>> # example 3: static graph
            >>> import paddle
            >>> import paddle.static as static

            >>> paddle.enable_static()

            >>> # create network
            >>> x = paddle.static.data(name="x", shape=[None, 224], dtype='float32')
            >>> z = paddle.static.nn.fc(x, 10)

            >>> place = paddle.CPUPlace()
            >>> exe = paddle.static.Executor(place)
            >>> exe.run(paddle.static.default_startup_program())
            >>> prog = paddle.static.default_main_program()
            >>> for var in prog.list_vars():
            ...     if list(var.shape) == [224, 10]:
            ...         tensor = var.get_value()
            ...         break

            >>> # save/load tensor
            >>> path_tensor = 'temp/tensor.pdtensor'
            >>> paddle.save(tensor, path_tensor)
            >>> load_tensor = paddle.load(path_tensor)

            >>> # save/load state_dict
            >>> path_state_dict = 'temp/model.pdparams'
            >>> paddle.save(prog.state_dict("param"), path_tensor)
            >>> load_state_dict = paddle.load(path_tensor)

        .. code-block:: python
            :name: code-example-4

            >>> # example 4: load program
            >>> import paddle

            >>> paddle.enable_static()

            >>> data = paddle.static.data(
            ...     name='x_static_save', shape=(None, 224), dtype='float32')
            >>> y_static = z = paddle.static.nn.fc(data, 10)
            >>> main_program = paddle.static.default_main_program()
            >>> path = "example/main_program.pdmodel"
            >>> paddle.save(main_program, path)
            >>> load_main = paddle.load(path)

        .. code-block:: python
            :name: code-example-5

            >>> # example 5: save object to memory
            >>> from io import BytesIO
            >>> import paddle
            >>> from paddle.nn import Linear
            >>> paddle.disable_static()

            >>> linear = Linear(5, 10)
            >>> state_dict = linear.state_dict()
            >>> byio = BytesIO()
            >>> paddle.save(state_dict, byio)
            >>> paddle.seed(2023)
            >>> tensor = paddle.randn([2, 3], dtype='float32')
            >>> paddle.save(tensor, byio)
            >>> byio.seek(0)
            >>> # load state_dict
            >>> dict_load = paddle.load(byio)

    rV   rÉ   rÊ   Úlatin1©ÚencodingrA   rL   r1  Nz%`paddle.load` can not parse the file:rô   )+r   r^   r:   Úisfilerž   ra   ÚUnpicklingErrorr   r   rÎ   rÏ   rÐ   rÑ   r   rb   r#   r$   r   r"   rI   rî   rý   r—   ÚgetattrrL   r–   r	  r-  r"  r   r®   r¯   r°   r±   rJ   r²   r   r  r  r   Úparse_from_stringr/   Ú_legacy_load)r:   r=   re   Úexception_typerl   Úload_resultrP   rL   Ú
msg_picklerÿ   r€   r´   Úprogram_desc_strÚprograms                 r   rb   rb   À  s¿  € ôJ ˜Ô¤"§'¡'§.¡.°Õ"6Ü# GÓ,ˆÜ×/Ñ/ˆðH	Ü" 4¨Ô.°!ô " $Ô'ÜŸ™¨Ò0Ü×(Ñ(×.Ñ.°!Ò3ä"3°D¸!Ó"<‘Kä"(§+¡+¨a¸(Ô"C�Kô ˜k¬4Ô0Ü"3°KÓ"@�Kà4¸ÑCØ)4Ø9ñ*ç™%›'ò*"™I˜C ô  *¨+°cÑ*:¼B¿J¹JÕGÜ3EØ$/°Ñ$4°f×6IÑ6Ió4" ¨CÑ 0ð (.×':Ó':¼wØ$/°Ñ$4°f¸bõ@"ð =A K°Ñ$4Õ$9ð*"ð !'× 6Ò 6Ø =ÀÑ Là +Ð,IÑ Jô '9Ø'¨×)<Ñ)<ó'™ô
 #5Ø# V×%8Ñ%8ó#�K÷S /ðX Ðô # 4Ñ3¨7Ñ3ˆàÐ÷Y /Ð.ûðZ ó 	ðÜ/°Ó5‘	�˜Ø•øððÜ 0°Ó 6‘I�F˜AØ×*Ò*Ü ŸJ™J›L˜ØŸ™¤F§O¡OÓ$5Ô6Ü!Ÿx™x¨¯©°Q«Ó8×8ä*Ô,Ü#6°vÓ#>×>Ø%Ÿøðð
Ü.¨t°TÔ:¸aØ/0¯v©v«xÐ,Ü&-×&?Ñ&?Ø 0ó'˜Gð $+÷ ;×:×:Ó:úÐ:øðÜ(ØCÀDÀ6ÈÐKóð úñ ;úÑ:úÓ:ð Ðûð?	úsó   ÁF5 ÁB5F)Ä
(F)Ä3F)ÅAF)ÆF5 Æ)F2Æ.F5 Æ5K4Æ<GÇK4ÇK&ÇA4I5ÉK&É	K/É
K4ÉI5É$K&É%K/É&K4É,I5É-K&É.K/É/K4É5K!É8KÊ'J>	Ê+	KÊ4K!Ê5K&Ê7K/Ê8K4Ê>KËKË
K!ËK	ËK!ËK&Ë$K/Ë/K4c                 ó(  — d }t        |«      }t        j                  j                  | «      st	        | «      rLt        | d«      5 }t        j                  |d¬«      }d d d «       t        |«      }|j                  sd|v r|d= |S t        | |«      \  }}|j                  €d}n|j                  }t        j                  j                  ||«      }t        j                  j                  |«      rt        ||«      }|S t        |«      }|S # 1 sw Y   Œ­xY w)NrV   r>  r?  rA   Ú	__model__)rž   r^   r:   rA  r   r   ra   rb   r   r–   r”   rY   r   r_   ro   r†   )r:   r=   rG  re   rl   rd   rY   Úmodel_file_paths           r   rE  rE  ¸  sÿ   € Ø€KÜ Ó(€Fä	‡w�w‡~�~�dÔÔ0°Ô6ä˜t TÔ*¨aÜ Ÿ+™+ a°(Ô;ˆK÷ +ä'¨Ó4ˆà×&Ò&Ø-°Ñ<àÐ9Ð:ð> Ðô9 9¸¸vÓFÑˆ
�Fà× Ñ Ð(Ø(‰Nà#×2Ñ2ˆNÜŸ'™'Ÿ,™, z°>ÓBˆä�7‰7�>‰>˜/Ô*ô EØ˜FóˆKð Ðô <¸JÓGˆKàÐ÷M +Ð*ús   ÁDÄD)r¦   F)r¦   )r¥   )=rá   rÖ   r^   ra   rÎ   r4   r0   Úcollections.abcr   ÚnumpyrI   r°   r   Úpaddle.baser   Úpaddle.base.frameworkr   r   r   r	   r
   r   r   Úio_utilsr   r   r   r   r   r   r   Ú__all__r   r   r?   rQ   ro   r†   r”   rž   r¢   rÝ   rå   rì   rð   rò   r÷   rú   rý   r  rÍ   r	  r  r"  r(  r-  r/  r6   r:  rb   rE  © r   r   Ú<module>rU     sñ   ðó Û Û 	Û Û 
Û Û Ý $ã ã õ Ý ÷÷ ñ ÷÷ ñ ð €ØÐ òó<ò~ò*,ò^ò`7òtò<ò.IòXò.ò<òòò
%ò"òò òF"
òJò2ò,ò*ò0
ó u/óp49ònuóp,r   