Tests and miscellaneous

fadel / pytorch_ema

Tiny PyTorch library for maintaining a moving average of a collection of parameters.

MIT License

404 stars 26 forks source link

Closed Linux-cpp-lisp closed 3 years ago

Linux-cpp-lisp commented 3 years ago

Hi @fadel,

Thank you for writing and publishing this library! It's great to have an implementation of this technique available.

This PR contains:

Some unit tests (pytest format)
Adding explicit install dependency on PyTorch
Type hints
A few little pieces of clean-up to satisfy flake8 after it ran automatically in my IDE :smile:
One fewer temporary buffer in update by using an in-place multiply

fadel commented 3 years ago

Thanks for the very nice contribution!