[Core] add support for gradient checkpointing in transformer_2d #5943
add support for gradient checkpointing in transformer_2d
d457826c
Merge branch 'main' into feat/grad-ckpt-transformer
edb7aa77
sayakpaul
merged
3f7c3511
into main 2 years ago
sayakpaul
deleted the feat/grad-ckpt-transformer branch 2 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub