ColossalAI/colossalai/utils/activation_checkpoint.py at c6930d8ddfc5e288375da8d0bb3f6a97d7f3f708 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-14 22:28:18 +00:00

Files

Frank Lee 2412429d54 [util] fixed activation checkpointing on torch 1.9 (#719 )

2022-04-12 09:35:45 +08:00

5.5 KiB

Raw Blame History

View Raw