Skip to content

Add ViT gradient checkpointing - #2058

Open
jaredrainsha wants to merge 6 commits into
lightly-ai:masterfrom
jaredrainsha:feat/2054-vit-grad-checkpointing
Open

Add ViT gradient checkpointing#2058
jaredrainsha wants to merge 6 commits into
lightly-ai:masterfrom
jaredrainsha:feat/2054-vit-grad-checkpointing

fix: keep block grads under reentrant checkpointing with frozen inputs

0c5094a
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs