aboutsummaryrefslogtreecommitdiff
AgeCommit message (Expand)Author
2021-11-17 handle non-trainable parametersLinux-cpp-lisp
2021-07-02Merge pull request #8 from Linux-cpp-lisp/context_manager...Context manager & README updatesSamuel Fadel
2021-07-02Merge pull request #7 from Linux-cpp-lisp/device_and_dtype...Device and dtype improvementsSamuel Fadel
2021-04-27README updatesAlby M
2021-04-21context manager for validationAlby M
2021-04-21IndentsAlby M
2021-04-21Add .to()Alby M
2021-04-21More state_dict testsAlby M
2021-04-21Casting and error checkingAlby M
2021-04-21Cast state dict to current device/dtypeAlby 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)...* linter * Add tests * Add restore test * Type hints * PyTorch dependency * One fewer temp bufferAlby M
2021-03-03Version bump and only store params requiring grad.v0.2Samuel Fadel
2021-03-03Updated README.md with store/restore functionality.Samuel Fadel
2021-03-03Minor 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-08Merge pull request #2 from Zehui-Lin/master...Add feature: store/restoreSamuel Fadel
2021-02-08minorZehui Lin
2021-01-21fix commentsZehui Lin
2021-01-21store and restoreZehui Lin
2021-01-15Add Feature: restorezehui-lin
2020-08-20Updated README.md with suggested install method.Samuel Fadel
2019-05-26Filter by requires_grad before zip() in update().Samuel Fadel
2019-04-16Initial commit.Samuel Fadel