ver217 7c6c427db1 [zero] trace states of fp16/32 grad and fp32 param (#571)

2022-03-31 16:26:54 +08:00

..

__init__.py

[memory] add model data tensor moving api (#503)

2022-03-24 14:29:41 +08:00

bucket_tensor_copy.py

[memory] set cuda mem frac (#506)

2022-03-24 16:57:13 +08:00

memory_monitor.py

Refactored docstring to google style

2022-03-29 17:17:47 +08:00

utils.py

[zero] trace states of fp16/32 grad and fp32 param (#571)

2022-03-31 16:26:54 +08:00