mmengine

mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00

History

[Feature] Enable bf16 in AmpOptimWrapper (#960 )

* support bf16 in AmpOptimWrapper

* add docstring

* modify docs

* add unittests for bf16 in AmpOptimWrapper

* fix type

* fix to pass ci

* fix ut skip logic to pass ci

* fix as comment

* add type hints

* fix docstring and add warning information

* remove check for pytorch>=1.6 in unittest

* modify unittest

* modify unittest

* remove torch.float32 && torch.float64 from valid dtypes

* fix as comments

* minor refine docstring

* fix unittest parameterized to pass CI

* fix unittest && add back torch.float32, torch.float64

2023-03-01 21:35:18 +08:00

_static

[Docs] Update two-column table wrapper (#726 )

2022-12-06 17:22:02 +08:00

_templates

[Docs] Make API of docs clearer (#469 )

2022-08-30 16:27:20 +08:00

advanced_tutorials

[Docs] Count FLOPs and parameters (#939 )

2023-02-21 21:16:18 +08:00

api

[Feature] Add NPUProfilerHook to profile performance in Ascend device (#925 )