Commit Graph

2 Commits (1927bc772672acd40df1abc8d6cac64294e85bc8)

Author SHA1 Message Date
RangiLyu 9a61b389e7
[Refactor] Add batch_idx to hook input. ()
* [Refactor] Add batch_idx to hook input.

* update
2022-03-29 11:40:38 +08:00
Yuan Liu 63a3af4f8c
[Feature]: Add optimizer hook ()
* [Feature]: Add optimizer hook

* [Fix]: Update docstring

* [Fix]: Add call with in UT
2022-03-01 17:42:15 +08:00