CoolFace
Apppublic

Oopstom/ReactSeq

sourceHugging Facelgpl-2.1updated 1y agoView on Hugging Face
6likes
rootonmtbin__pycache__average_models.cpython-38.pyc
average_models.cpython-38.pyc15 linesDownload Raw Back to __pycache__
1U

2b�Nes�@s4ddlZddlZddd�Zdd�Zedkr0e�dS)	�NFcCs d}d}d}d}t|�D]�\}}tj|dd�}|d}	|d}3|r~|	��D]\}}|��|	|<qJ|4��D]\}}|��|5|<qh|dkr�|d|d}}|	}|6}q|��D]*\}}||�|��|	|��|d�q�|��D]*\}}||�|��|7|��|d�q�q||d||d	�}
|
S)8N�cpu)Zmap_location�model�	generatorr�vocab�opt�)rrZoptimrr)�	enumerate�torch�load�items�floatZmul_Zadd_Zdiv_)Zmodel_files�fp32rrZ	avg_modelZ
avg_generator�iZ9model_file�mZ
model_weightsZgenerator_weights�k�v�final�r�]/home/zhangwei/anaconda3/envs/opennmt3/lib/python3.8/site-packages/onmt/bin/average_models.py�average_modelss8$&�rcCsltjdd�}|jdddddd�|jd	d10ddd�|jd
dddd�|��}t|j|j�}t�||j	�dS)N�)�descriptionz-modelsz-m�+TzList of models)�nargs�required�helpz-outputz-ozOutput file)rrz-fp32z-f�11store_truezCast params to float32)�actionr)12�argparse�ArgumentParser�add_argument�13parse_argsr�modelsr
r	�save�output)�parserrrrrr�main,s$��r&�__main__)F)rr	rr&�__name__rrrr�<module>s1415&