You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

CsRic 9623ec1b02 [NFC] polish colossalai/amp/naive_amp/_utils.py code style (#1816) ... * [NFC] polish colossalai/nn/metric/accuracy_2p5d.py code style (#1714) * [NFC] polish colossalai/zero/sharded_param/__init__.py code style * [NFC] polish colossalai/amp/naive_amp/_utils.py code style Co-authored-by: shenggan <csg19971016@gmail.com> Co-authored-by: ric <mkkt_bkkt@mail.ustc.edu.cn>		2 years ago
..
grad_scaler	[doc] improved docstring in the amp module (#857)	3 years ago
__init__.py	[doc] improved docstring in the amp module (#857)	3 years ago
_fp16_optimizer.py	[hotfix]different overflow status lead to communication stuck. (#1175)	2 years ago
_utils.py	[NFC] polish colossalai/amp/naive_amp/_utils.py code style (#1816)	2 years ago
naive_amp.py	[amp] included dict for type casting of model output (#1102)	3 years ago