mirror of https://github.com/hpcaitech/ColossalAI
![]() * [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. |
||
---|---|---|
.. | ||
__init__.py | ||
ckpt_solver_chen.py |