Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Ycr's avatar
    [Feature] Support revise_keys in load_checkpoint(). (#829) · e076c8b0
    Ycr authored 
    * Simplified the code.

* Improved chkpt compatibility.

* One may modify the checkpoint via adding keywords.

* Tiny.

* Following reviewer's suggestion.

* Added unit_test.

* Fixed.

* Modify the state_dict  with  construction.

* Added test.

* Modified。

* Mimimalised the modification.

* Added the docstring.

* Format.

* Improved.

* Tiny.

* Temp file.

* Added assertion.

* Doc string.

* Fixed.
    e076c8b0
test_checkpoint.py 11.9 KB