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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-15 14:47:16 +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