Csatv2 contribution#2627

Merged

rwightman merged 18 commits intomainfrom

csatv2-gusdlf93

Dec 12, 2025

Commits on Dec 9, 2025

Upload CSATv2

gusdlf93
authored and
rwightman
committed
Update csatv2.py

gusdlf93
authored and
rwightman
committed
Update csatv2.py

gusdlf93
authored and
rwightman
committed
Add files via upload

gusdlf93
authored and
rwightman
committed
Add files via upload

gusdlf93
authored and
rwightman
committed
Delete csatv2.py

gusdlf93
authored and
rwightman
committed
Add files via upload

gusdlf93
authored and
rwightman
committed
Pass the Test

gusdlf93
authored and
rwightman
committed
Some rework of csatv2 to better fit timm norms, re-use existing layers where possible
rwightman
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
rwightman
committed
Weird mix of camel/snake conv naming -> snake case
rwightman
committed

Commits on Dec 10, 2025

Commits on Dec 11, 2025

Another round of consistency changes for csatv2, make stage building dynamic for other network shapes, allow drop path option for transformer blocks.
rwightman
committed