xdecoder/Instruct-X-Decoder
163
1U
2 E$�c# � @ s* d dl Z d dlmZ e �e�Zdd� ZdS )� N)�is_main_processc
C s" t | �� �}t |�� �}i }g }g }g }|D ]�}| | } ||kr�|| }3| j|4jkr�|5||< |�|�|�� |�d�|| j|6j�� q�|�d�|| j|7j�� q,|�d�|| j�� q,t� �r |D ]} t� | � q�|D ]} t�8| � q�|D ] } t�9d�| || j�� q�|D ]} t�10| � �q |S )Nz-Loaded {}, Model Shape: {} <-> Ckpt Shape: {}z2*UNMATCHED* {}, Model Shape: {} <-> Ckpt Shape: {}z *UNLOADED* {}, Model Shape: {}z$UNUSED$ {}, Ckpt Shape: {}) �sorted�keys�shape�pop�index�append�formatr �logger�info�warning)
�model_state_dictZckpt_state_dictZ11model_keysZ ckpt_keysZ result_dictsZ matched_logZ
unmatched_logZ unloaded_logZ model_keyZ model_weightZ ckpt_weightr �key� r �,/data/arXiv/demo/Demo/utils/model_loading.py� align_and_update_state_dicts
s4 r )�logging�utils.distributedr � getLogger�__name__r12 r r r r r �<module> s 13