mmengine

Author	SHA1	Message	Date
luomaoling	5b9a1544b0	[Feature] Add torch_npu optimizer (#1079 )	2023-04-21 15:15:10 +08:00
Qian Zhao	0d25625ba2	[Feature] Support torch.compile since PyTorch2.0 (#976 ) * enable compile configurations to support torch.compile in Runner * enable compilation in train, val and test * fix as comments * add docstring to illustrate usage * minor refine error message * add unittests * fix ut skip * add logging message to inform users * compile `train_step`, `val_step`, `test_step` instead * fix as comments * revert to compile `train_step` only due to pt2 issue * add documentation about torch.compile	2023-03-12 18:26:43 +08:00
Qian Zhao	2ed8e343a0	[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
Zaida Zhou	4749682885	[Docs] Move examples to a new directory (#911 ) * [Docs] Move examples to a new directory * fix typo	2023-02-07 17:03:35 +08:00