yingtongxiong 363275b500 add memory print		2023-10-25 14:31:00 +08:00
..
apis	initial commit	2023-07-06 12:55:23 +08:00
core	support model activation checkpoint	2023-10-24 16:13:52 +08:00
data	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
initialize	feat(initialize/launch.py): refactor config for fstp	2023-10-20 15:59:40 +08:00
model	add memory print	2023-10-25 14:31:00 +08:00
moe	feat(moe): add moe module (#182)	2023-09-27 15:54:53 +08:00
monitor	doc(monitor): add light monitoring doc (#352)	2023-09-25 19:28:09 +08:00
solver	fix accumulate grads bug	2023-10-23 16:17:57 +08:00
train	support tflops computation and generate test py files	2023-10-24 17:54:26 +08:00
utils	support model activation checkpoint	2023-10-24 16:13:52 +08:00
__init__.py	initial commit	2023-07-06 12:55:23 +08:00