mirror of https://github.com/hpcaitech/ColossalAI
e7383f578b
* [fx] polish ckpt_test. * [fx] add rules to linearize computation graphs for searching. * [fx] remove chen_sqrt for sake of simplicity * [fx] fix inconsistencies. |
||
---|---|---|
.. | ||
test_ckpt_torchvision.py |