Skip to content

Commit b0cb744

Browse files
rwightmanrwightman
committed
Add grad_checkpointing. Remap head to use NormMlpClassifierHead and final norm is 2d so we can disable pooling if desired. Still inconsistent line endings
1 parent dad1ca1 commit b0cb744

File tree

1 file changed

+764
-752
lines changed

1 file changed

+764
-752
lines changed

0 commit comments

Comments
 (0)