Files
ColossalAI/colossalai
Super Daniel 3b26516c69 [fx] add vanilla activation checkpoint search with test on resnet and densenet (#1433)
* [fx] activation checkpointing using Chen strategies.

* [fx] add test for ckpt_solver_chen

* [fx] add vanilla activation checkpoint search with test on resnet and densenet

* [fx] add vanilla activation checkpoint search with test on resnet and densenet

* [fx] add a namespace code for solver_chen.
2022-08-11 15:46:39 +08:00
..
2022-08-11 13:43:24 +08:00
2022-06-07 17:21:11 +08:00
2022-05-26 08:45:08 +08:00