tf keras,取相同型号的平均重量
modelaverage的Python项目详细描述
模型平均值
` model average是一个pip包,它根据[keras模型中的平均权重](https://stackoverflow.com/questions/48212110/average weights in keras models)的启发,生成“相同模型”中模型权重的平均权重。我创建这个pip包是为了使用像“kubernetes”这样的分布式计算环境。用法
pip install modelaverage
或git clone https://github.com/graykode/modelaverage && python setup.py install
- 使用函数
average(modellist)
- 模型列表:模型文件名的列表。
- 返回:平均重量模型
示例
请参见示例
importtensorflowastffrommodelaverageimportaveragemodellist=['models/mnist1.h5','models/mnist2.h5','models/mnist3.h5','models/mnist4.h5','models/mnist5.h5','models/mnist6.h5','models/mnist7.h5','models/mnist8.h5','models/mnist9.h5']averaged_model=average(modellist)forwinaveraged_model.get_weights():print(w.shape)
作者
- 姓名:郑泰桓(@graykode)
- 电子邮件:nlkey2022@gmail.com