HELSON e5ea3fdeef [gemini] add GeminiMemoryManger (#832) ...

* refactor StatefulTensor, tensor utilities

* add unitest for GeminiMemoryManager

2022-04-24 13:08:48 +08:00

..

test_checkpoint

[test] added missing decorators to model checkpointing tests

2022-04-12 11:08:15 +08:00

test_activation_checkpointing.py

[hotfix] fix bugs in testing (#659)

2022-04-02 21:58:47 +08:00

test_commons.py

[gemini] add GeminiMemoryManger (#832)

2022-04-24 13:08:48 +08:00

test_memory.py

[utils] correct cpu memory used and capacity in the context of multi-process (#726)

2022-04-12 14:57:54 +08:00

test_pipelinable.py

[pipelinable]use pipelinable context to initialize non-pipeline model (#816)

2022-04-24 13:03:12 +08:00

test_zero_gradient_clippling.py

[test] refactored with the new rerun decorator (#763)

2022-04-15 00:33:04 +08:00