Age | Commit message (Expand) | Author |
---|---|---|
2021-04-21 | Cast state dict to current device/dtype | Alby M |
2021-04-21 | State dict support (#6) | Alby M |
2021-04-20 | Option to keep parameters reference in `ExponentialMovingAverage` (#5) | Alby M |
2021-04-20 | Tests and miscellaneous (#4)...* linter * Add tests * Add restore test * Type hints * PyTorch dependency * One fewer temp buffer | Alby M |
2021-03-03 | Version bump and only store params requiring grad.v0.2 | Samuel Fadel |
2021-03-03 | Minor updates to docstrings and code organization....* Docstrings using the same imperative verb form * Use list comprehension when collecting params with store() | Samuel Fadel |
2021-02-08 | minor | Zehui Lin |
2021-01-21 | fix comments | Zehui Lin |
2021-01-21 | store and restore | Zehui Lin |
2021-01-15 | Add Feature: restore | zehui-lin |
2019-05-26 | Filter by requires_grad before zip() in update(). | Samuel Fadel |
2019-04-16 | Initial commit. | Samuel Fadel |