LuGY c44d797072 [docs] updatad docs of hybrid adam and cpu adam (#552)		2022-03-30 18:14:59 +08:00
..
amp	Refactored docstring to google style	2022-03-29 17:17:47 +08:00
builder	Refactored docstring to google style	2022-03-29 17:17:47 +08:00
communication	Refactored docstring to google style	2022-03-29 17:17:47 +08:00
context	Refactored docstring to google style	2022-03-29 17:17:47 +08:00
engine	[zero] hijack p.grad in sharded model (#554)	2022-03-30 18:14:50 +08:00
kernel	[cuda] modify the fused adam, support hybrid of fp16 and fp32 (#497)	2022-03-25 14:15:53 +08:00
logging	Refactored docstring to google style	2022-03-29 17:17:47 +08:00
nn	[docs] updatad docs of hybrid adam and cpu adam (#552)	2022-03-30 18:14:59 +08:00
registry	Refactored docstring to google style	2022-03-29 17:17:47 +08:00
testing	[test] fixed rerun_on_exception and adapted test cases (#487)	2022-03-25 17:25:12 +08:00
trainer	Refactored docstring to google style	2022-03-29 17:17:47 +08:00
utils	[zero] dump memory stats for sharded model (#548)	2022-03-30 09:38:44 +08:00
zero	[zero] hijack p.grad in sharded model (#554)	2022-03-30 18:14:50 +08:00
__init__.py	…
constants.py	…
core.py	…
global_variables.py	…
initialize.py	Refactored docstring to google style	2022-03-29 17:17:47 +08:00