Hongxin Liu a15ab139ad [plugin] support get_grad_norm (#6115)		2024-11-05 18:12:47 +08:00
..
grad_scaler	[npu] change device to accelerator api (#5239)	2024-01-09 10:20:05 +08:00
mixed_precision_mixin	[npu] change device to accelerator api (#5239)	2024-01-09 10:20:05 +08:00
__init__.py	[legacy] clean up legacy code (#4743)	2023-09-18 16:31:06 +08:00
mixed_precision_optimizer.py	[plugin] support get_grad_norm (#6115)	2024-11-05 18:12:47 +08:00