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