Skip to content

ModelSoups for Tensorflow2 and Torch

License

Notifications You must be signed in to change notification settings

dreamfireyu/ModelSoups

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Model Soups

This is an implementation of "Model Soups" on Keras, Tensorflow and Torch.

The implementation is based on paper[1].

Method

  • Uniform Soup
  • Greedy Soup

Requirements

  • Python 3
  • Tensorflow2 or Torch

Reference

  1. Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference time, Mitchell Wortsman, Gabriel Ilharco, Samir Yitzhak Gadre, Rebecca Roelofs, Raphael Gontijo-Lopes, Ari S. Morcos, Hongseok Namkoong, Ali Farhadi, Yair Carmon, Simon Kornblith, Ludwig Schmidt, https://arxiv.org/abs/2203.05482v1

Contributor

About

ModelSoups for Tensorflow2 and Torch

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 81.1%
  • Python 18.9%