API for getting intermediate image and text features, forward_intermediates()#1035

Merged

rwightman merged 8 commits intomainfrom

Mar 1, 2025

Commits on Feb 22, 2025

Commits on Feb 24, 2025

Add forward_intermediates to ResNet tower, fix several issues, test more cases.
rwightman
committed
Fix corner case with intermediates_only and normalization active
rwightman
committed

Commits on Feb 25, 2025

Reorganize, rename some output args. Split image/text extra tokens. Add partial impl to CoCa model.
rwightman
committed

Commits on Feb 26, 2025

Commits on Feb 27, 2025

key check on normalize for CoCa, use_reentrant=False checkpointing for CoCa
rwightman
committed