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