Age | Commit message (Collapse) | Author | |
---|---|---|---|
2021-04-21 | More state_dict tests | 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 |