1. 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
  2. 14 Jan, 2021 1 commit
  3. 08 Jan, 2021 1 commit
  4. 22 Dec, 2020 1 commit
  5. 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