Super Daniel d40a9392ba [fx] fix the false interpretation of algorithm 3 in https://arxiv.org/abs/1604.06174. (#1446) ... * [fx] modify the calculation of node_size in MetaInfoProp for activation checkpointing usages * [fx] modify the calculation of node_size in MetaInfoProp for activation checkpointing usages * [fx] modify the calculation of node_size in MetaInfoProp for activation checkpointing usages * [fx] activation checkpointing using Chen strategies. * [fx] add test for ckpt_solver_chen * mend * [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. * [fx] fix the false interpretation of algorithm 3 in https://arxiv.org/abs/1604.06174. * [fx] fix lowercase naming conventions.		2022-08-12 11:28:50 +08:00
..
__init__.py	[fx] add vanilla activation checkpoint search with test on resnet and densenet (#1433)	2022-08-11 15:46:39 +08:00
ckpt_solver_chen.py	[fx] fix the false interpretation of algorithm 3 in https://arxiv.org/abs/1604.06174. (#1446)	2022-08-12 11:28:50 +08:00