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.

HELSON 077a5cdde4 [zero] fix gradient clipping in hybrid parallelism (#2521) ... * [zero] fix gradient clipping in hybrid parallelism * [testing] change model name to avoid pytest warning * [hotfix] fix unit testing		2 years ago
..
apex_amp	[NFC] polish colossalai/amp/apex_amp/__init__.py code style (#1853)	2 years ago
naive_amp	[zero] fix gradient clipping in hybrid parallelism (#2521)	2 years ago
torch_amp	[NFC] polish colossalai/amp/torch_amp/torch_amp.py code style (#2290)	2 years ago
__init__.py	[refactor] zero directory (#724)	3 years ago
amp_type.py	Develop/experiments (#59)	3 years ago