aboutsummaryrefslogtreecommitdiff
path: root/tests/test_ema.py
AgeCommit message (Collapse)Author
2021-04-21context manager for validationAlby M
2021-04-21Add .to()Alby M
2021-04-21More state_dict testsAlby M
2021-04-21State dict support (#6)Alby M
2021-04-20Option to keep parameters reference in `ExponentialMovingAverage` (#5)Alby M
2021-04-20Tests and miscellaneous (#4)Alby M
* linter * Add tests * Add restore test * Type hints * PyTorch dependency * One fewer temp buffer