Jiarui Fang 49832b2344 [refactory] add nn.parallel module (#1068)		2022-06-06 15:34:41 +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	[refactory] add nn.parallel module (#1068)	2022-06-06 15:34:41 +08:00
test_zero_gradient_clippling.py	[test] refactored with the new rerun decorator (#763)	2022-04-15 00:33:04 +08:00