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
..
debug_tricks.md	[Docs] Add a document about debug tricks (#938)	2023-02-21 21:40:35 +08:00
epoch_to_iter.md	[Docs] Add the document for the transition between IterBasedTraining and EpochBasedTraining (#926)	2023-02-21 21:12:38 +08:00
model_analysis.md	[Docs] Count FLOPs and parameters (#939)	2023-02-21 21:16:18 +08:00
resume_training.md	[Docs] Move examples to a new directory (#911)	2023-02-07 17:03:35 +08:00
save_gpu_memory.md	[Docs] Move examples to a new directory (#911)	2023-02-07 17:03:35 +08:00
set_interval.md	[Docs] how to set the interval parameter (#917)	2023-02-14 11:42:49 +08:00
set_random_seed.md	[Docs] Introduce how to set random seed (#914)	2023-02-13 16:14:48 +08:00
speed_up_training.md	[Feature] Enable bf16 in AmpOptimWrapper (#960)	2023-03-01 21:35:18 +08:00