mmengine

Commit Graph

Author	SHA1	Message	Date
Mashiro	abe56651db	[Fix] Make autocast compatible with mps (#587 ) * [Fix] Make autocast compatible with mps * Enhance unit test * fix unit test * clean the code * fix unit test	2022-10-18 18:03:17 +08:00
Zaida Zhou	7e1d7af2d9	[Refactor] Refactor code structure (#395 ) * Rename data to structure * adjust the way to import module * adjust the way to import module * rename Structure to Data Structures in docs api * rename structure to structures * support using some modules of mmengine without torch * fix circleci config * fix circleci config * fix registry ut * minor fix * move init method from model/utils to model/weight_init.py * move init method from model/utils to model/weight_init.py * move sync_bn to model * move functions depending on torch to dl_utils * format import * fix logging ut * add weight init in model/__init__.py * move get_config and get_model to mmengine/hub * move log_processor.py to mmengine/runner * fix ut * Add TimeCounter in dl_utils/__init__.py	2022-08-24 19:14:07 +08:00
Mashiro	59b0ccfe6f	[Fix] Fix pytorch version compatibility of autocast (#339 ) * fix unit test of autocast * fix compatiblity of unit test of optimizerwrapper * clean code * fix as comment * fix docstring	2022-06-29 20:30:53 +08:00
Mashiro	312f264ecd	[Feature] Add autocast wrapper (#307 ) * add autocast wrapper * fix docstring * fix docstring * fix compare version * fix unit test * fix incompatible arguments * fix as comment * fix unit test * rename auto_cast to autocast	2022-06-22 19:49:20 +08:00

4 Commits (bda92b49e751c721c11bb15623dabc7f7a9517e7)