https://github.com/graykode/modelaverage

tf-keras, make the average of model weight in same model.
https://github.com/graykode/modelaverage

Last synced: 10 months ago
JSON representation

tf-keras, make the average of model weight in same model.

• Host: GitHub
• URL: https://github.com/graykode/modelaverage
• Owner: graykode
• License: mit
• Created: 2019-03-12T09:30:53.000Z (over 7 years ago)
• Default Branch: master
• Last Pushed: 2019-03-12T15:51:02.000Z (over 7 years ago)
• Last Synced: 2025-07-06T23:49:35.953Z (12 months ago)
• Language: Python
• Homepage:
• Size: 10.7 MB
• Stars: 3
• Watchers: 3
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
## modelaverage



    

    



`modelaverage` is a pip package that calculates the average value of the same model, inspired by [Average weights in keras models](https://stackoverflow.com/questions/48212110/average-weights-in-keras-models). I created it to train each mini-batch in one container as distributed computing environment, `kubernetes`.

![](average.jpg)

## Usage

1. `pip install modelaverage` or `git clone https://github.com/graykode/modelaverage && python setup.py install`

2. using function `average(modellist)`

   - modellist : list of model file names. type should be `list`

     > averaged_model = average(['mnist1.h5', 'mnist2.h5',....])

   - return : input models of averaged weight

## Example

Please see [example](https://github.com/graykode/modelaverage/blob/master/example/example.py).

```python

import tensorflow as tf

from modelaverage import average

modellist = ['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)

for w in averaged_model.get_weights():

    print(w.shape)

```

## Author

- Name : Tae Hwan Jung(@graykode)

- Email : nlkey2022@gmail.com