6 Commits (34b552b881a17c0b63bd86854e843e844ade5b19)

Author	SHA1	Message	Date
Miao Zheng	a4c3702628	add initializers and BaseModule for unified parameter initialization (#780) ... * 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	2021-02-07 11:13:57 +08:00
Haian Huang(深度眸)	96892bdc42	Refactor _load_checkpoint fn (#790) ... * 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	2021-01-28 15:09:19 +08:00
BigBigDream	8e3a801596	fix mmcv ci for parrots (#782) ... * fix mmcv ci for parrots * fix mmcv ci * fix lint	2021-01-14 14:31:57 +08:00
Qiaofei Li	3d6032c6db	Add dataset classes name info to meta for saving ckpt (#776) ... * add CLASSES to meta info * Update checkpoint.py * add unit test for CLASSES name * clean up the tmp folder * use tempfile to clean up temp folder	2021-01-08 16:40:43 +08:00
LXXXXR	09661d9754	[Feature] support upload and download checkpoint from pavimodelcloud (#725) ... * support upload and download checkpoint from pavimodelcloud * try import pavi * add unittest	2020-12-22 12:59:07 +08:00
Harry	5704613e28	Remove all module wrapper's module when saving checkpoint (#399) ... * 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	2020-07-08 23:20:22 +08:00