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_logitsJun 7, 2024