Skip to content

Fix consistency, testing for forward_head w/ pre_logits, reset_classifier, models with pre_logits size != unpooled feature size#2195

Merged
rwightman merged 2 commits intomainfrom
refactor_pre_logits
Jun 7, 2024
Merged

Fix consistency, testing for forward_head w/ pre_logits, reset_classifier, models with pre_logits size != unpooled feature size#2195
rwightman merged 2 commits intomainfrom
refactor_pre_logits

Commits