Age | Commit message (Collapse) | Author | |
---|---|---|---|
2021-04-21 | Casting and error checking | Alby M | |
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) | Alby M | |
* linter * Add tests * Add restore test * Type hints * PyTorch dependency * One fewer temp buffer | |||
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. | Samuel Fadel | |
* Docstrings using the same imperative verb form * Use list comprehension when collecting params with store() | |||
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 | |