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