3 Commits

Author	SHA1	Message	Date
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