1. 03 Mar, 2021 1 commit
    • 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
  2. 07 Feb, 2021 1 commit
    • Miao Zheng's avatar
      add initializers and BaseModule for unified parameter initialization (#780) · a4c37026
      Miao Zheng authored
      * add initializers and BaseModule for unified parameter initialization
      
      * fix circle import
      
      * bug fix
      
      * add is_init flag in BaseModule
      
      * fix docstring
      
      * sort import and fix doc format
      
      * fix bug
      
      * fix docformat and double quote string
      
      * fix import sort
      
      * import sort
      
      * sort import
      
      * revise according to comments
      
      * fix doc format
      
      * revise according to comments
      
      * revise import and fix typo
      
      * polish code
      
      * revise minors
      
      * revice minors
      
      * revise apply function
      
      * revise bias initialization with probability
      
      * add type test for bias_prob
      
      * revise minors
      a4c37026
  3. 28 Jan, 2021 1 commit
    • Haian Huang(深度眸)'s avatar
      Refactor _load_checkpoint fn (#790) · 96892bdc
      Haian Huang(深度眸) authored
      * Refactor _load_checkpoint fn
      
      * Update _load_checkpoint fn
      
      * Update docs str and add unit test
      
      * Fix unit test
      
      * Fix lint
      
      * Add comment and Optimize function
      
      * Fix docs str
      
      * Update load_ckpt and fix doc str
      
      * Update doc str and add sort unit test
      
      * Update and fix unit test
      
      * Fix unit test
      
      * Update and add unit test
      
      * Fix openmmlab prefix error
      96892bdc
  4. 14 Jan, 2021 1 commit
  5. 08 Jan, 2021 1 commit
  6. 22 Dec, 2020 1 commit
  7. 08 Jul, 2020 1 commit
    • Harry's avatar
      Remove all module wrapper's module when saving checkpoint (#399) · 5704613e
      Harry authored
      * fix: remove all module wrapper when saving checkpoint
      
      * refactor: move position of if
      
      * docs: add docstring
      
      * refactor: add _save_to_state_dict from official torch
      
      * docs: modify docstring of _save_to_state_dict
      
      * docs: modify docstring
      
      * feat: add unittest
      
      * feat: add DataParallel to unittest
      
      * fix: a bug when model has batchnorm
      
      * docs: update docstring
      5704613e