aboutsummaryrefslogtreecommitdiff
path: root/torch_ema
AgeCommit message (Collapse)Author
2021-03-03Minor 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-08minorZehui Lin
2021-01-21fix commentsZehui Lin
2021-01-21store and restoreZehui Lin
2021-01-15Add Feature: restorezehui-lin
2019-05-26Filter by requires_grad before zip() in update().Samuel Fadel
2019-04-16Initial commit.Samuel Fadel